| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | one_hot_tests.cc | 36 float off_value = -1; in TEST_F() local
57 float off_value = -1; in TEST_F() local
78 float off_value = -1; in TEST_F() local
101 float off_value = -1; in TEST_F() local
124 float off_value = 0; in TEST_F() local
145 float off_value = 0; in TEST_F() local
167 float off_value = 0; in TEST_F() local
188 float off_value = -1; in TEST_F() local
211 float off_value = -2; in TEST_F() local
234 float off_value = -2; in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | one_hot_impl.cu | 20 …id OneHotKernel(size_t size, const S *indices, size_t depth, const T *on_value, const T *off_value, in OneHotKernel()
41 void OneHot(const S *indices, size_t depth, const T *on_value, const T *off_value, size_t left_dim_… in OneHot()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | one_hot_base.cc | 210 const auto *off_value = reinterpret_cast<float *>(off_value_tensor->data()); in InitOnOffValueForFourInputs() local
217 const auto *off_value = reinterpret_cast<float16_t *>(off_value_tensor->data()); in InitOnOffValueForFourInputs() local
|
| /third_party/mindspore/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/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | one_hot_fp32.c | 20 int OneHotToFp32(const int *indices, float on_value, float off_value, float *output, in OneHotToFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/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/tests/ut/cpp/ops/ |
| D | test_ops_one_hot.cc | 42 …auto off_value = TensorConstructUtils::CreateZerosTensor(kNumberTypeFloat32, std::vector<int64_t>{… in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | one_hot_cpu_kernel.cc | 57 auto off_value = reinterpret_cast<float *>(inputs[2]->addr)[0]; in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | one_hot_gpu_kernel.h | 45 const T *off_value = GetDeviceAddress<T>(inputs, 2); in Launch() local
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | convert_const_input_test.py | 36 off_value = Tensor(0.0, mstype.float32) variable
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | basic.py | 650 def __init__(self, axis=-1, depth=1, on_value=1.0, off_value=0.0, dtype=mstype.float32): argument
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_nn_ops.py | 825 def bprop(indices, depth, on_value, off_value, out, dout): argument
|