Home
last modified time | relevance | path

Searched refs:kDataTypeFloat16 (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dstrided_slice_fp32.c121 if (param->data_type != kDataTypeFloat && param->data_type != kDataTypeFloat16) { in DoStridedSlice()
159 } else if (param->data_type == kDataTypeFloat16) { in DoStridedSlice()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dstrided_slice_fp16_grad.cc46 param_->data_type = kDataTypeFloat16; in Init()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dstrided_slice.cc203 param_->data_type = kDataTypeFloat16; in NormalRun()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dop_base.h247 kDataTypeFloat16, enumerator