| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | one_hot_tests.cc | 35 float on_value = 1; in TEST_F() local
56 float on_value = 1; in TEST_F() local
77 float on_value = 1; in TEST_F() local
100 float on_value = 1; in TEST_F() local
123 float on_value = 2; in TEST_F() local
144 float on_value = 2; in TEST_F() local
166 float on_value = 2; in TEST_F() local
187 float on_value = 1; in TEST_F() local
210 float on_value = 2; in TEST_F() local
233 float on_value = 2; in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/ |
| D | test_onehot.py | 25 def onehot_forward_func(indices, depth, on_value, off_value, axis): argument
30 def onehot_backward_func(indices, depth, on_value, off_value, axis): argument
35 def onehot_vmap_func(indices, depth, on_value, off_value, axis): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | one_hot_impl.cu | 22 __global__ void OneHotKernel(size_t size, const S *indices, size_t depth, const T *on_value, const … in OneHotKernel()
43 cudaError_t OneHot(const S *indices, size_t depth, const T *on_value, const T *off_value, size_t le… in OneHot()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_one_hot.py | 32 def construct(self, indices, depth, on_value, off_value): argument
74 def construct(self, indices, depth, on_value, off_value): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_one_hot.py | 16 def construct(self, indices, on_value, off_value): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/auto_parallel/ |
| D | onehot_model_parallel.py | 49 def __init__(self, axis=-1, depth=1, on_value=1.0, off_value=0.0, strategy=None): argument
102 def __init__(self, batch_size, classes, on_value=1.0, off_value=0.0, axis=None, strategy=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | one_hot_infer.c | 32 const TensorC *on_value = inputs[2]; in OneHotInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | one_hot_fp16.c | 19 int OneHotToFp16(const int *indices, float16_t on_value, float16_t off_value, float16_t *output, in OneHotToFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | one_hot_fp32.c | 20 int OneHotToFp32(const int32_t *indices, float on_value, float off_value, float *output, in OneHotToFp32()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | one_hot_ext.cc | 29 … const Int64ImmPtr &num_classes, const BaseTensorPtr &on_value, in OneHotExtAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_onehot.cc | 53 …auto on_value = std::make_shared<abstract::AbstractTensor>(param.on_value_type, param.on_value_sha… in TEST_P() local
|
| D | test_ops_onehot_ext.cc | 53 …auto on_value = std::make_shared<abstract::AbstractTensor>(param.on_value_type, param.on_value_sha… in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | one_hot_gpu_kernel.cc | 101 const T *on_value = GetDeviceAddress<T>(inputs, on_value_idx); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | one_hot.cc | 103 auto on_value = reinterpret_cast<T *>(ctx.Input(2)->GetData()); in OneHotCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | one_hot_cpu_kernel.cc | 125 auto on_value = GetDeviceAddress<OD>(inputs, kIndex2)[0]; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | convert_const_input_test.py | 35 on_value = Tensor(1.0, mstype.float32) variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | basic.py | 865 def __init__(self, axis=-1, depth=1, on_value=1.0, off_value=0.0, dtype=mstype.float32): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | array_func.py | 657 def one_hot(indices, depth, on_value=1, off_value=0, axis=-1): argument
|