* Add one-hot to Relay * topi implementation * Working * add topi test * Add TF test * Fix check * fix linting issues * fix documentation * Fix documentation * Add support for on_value, off_value, axis, dtype * Add full support for axis * Fix compute and update test_forward * Move on_value and off_value to inputs * Add topi test * Update tests * Update docs * Fix style * re-enable tests * Add one_hot to mxnet converter
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
__init__.py | Loading commit data... | |
batch_matmul.py | Loading commit data... | |
bilinear_resize_python.py | Loading commit data... | |
conv2d_hwcn_python.py | Loading commit data... | |
conv2d_nchw_python.py | Loading commit data... | |
conv2d_nhwc_python.py | Loading commit data... | |
conv2d_transpose_nchw_python.py | Loading commit data... | |
deformable_conv2d_nchw_python.py | Loading commit data... | |
depthwise_conv2d_python.py | Loading commit data... | |
dilate_python.py | Loading commit data... | |
gather_nd_python.py | Loading commit data... | |
l2_normalize_python.py | Loading commit data... | |
lrn_python.py | Loading commit data... | |
one_hot.py | Loading commit data... | |
pool_grad_python.py | Loading commit data... | |
reorg_python.py | Loading commit data... | |
roi_align_python.py | Loading commit data... | |
roi_pool_python.py | Loading commit data... | |
sequence_mask_python.py | Loading commit data... | |
slice_axis_python.py | Loading commit data... | |
softmax_python.py | Loading commit data... | |
strided_slice_python.py | Loading commit data... | |
upsampling_python.py | Loading commit data... |