| /third_party/mindspore/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/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | one_hot_impl.cu | 20 __global__ void OneHotKernel(size_t size, const S *indices, size_t depth, const T *on_value, const … 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 | 185 const auto *on_value = reinterpret_cast<float *>(on_value_tensor->data()); in InitOnOffValueForFourInputs() local
192 const auto *on_value = reinterpret_cast<float16_t *>(on_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/infer/ |
| D | one_hot_infer.c | 32 const TensorC *on_value = inputs[2]; in OneHotInferShape() local
|
| /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/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/tests/ut/cpp/ops/ |
| D | test_ops_one_hot.cc | 41 …auto on_value = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1}… in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | one_hot_cpu_kernel.cc | 56 auto on_value = reinterpret_cast<float *>(inputs[1]->addr)[0]; in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | one_hot_gpu_kernel.h | 44 const T *on_value = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| /third_party/mindspore/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/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
|