Home
last modified time | relevance | path

Searched defs:weight_type (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dbce_with_logits_loss_kernel.cc61 auto weight_type = inputs[kIndex2]->GetType(); in Resize() local
108 auto weight_type = inputs[kIndex2]->GetType(); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/op/
Dfull_connection_nnapi.cc46 auto weight_type = in_tensors_.at(1).DataType(); in AddOpToNNAPIModel() local
Dconv_transpose_nnapi.cc66 auto weight_type = in_tensors_.at(1).DataType(); in AddOpToNNAPIModel() local
Dconv_nnapi.cc126 auto weight_type = in_tensors_.at(1).DataType(); in AddOpToNNAPIModel() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dgroup_norm.cc92 const auto &weight_type = input_args[kInputIndex2]->GetType(); in InferType() local
116 const auto &weight_type = weight_tensor->Dtype(); in InferType() local
Dprelu.cc44 auto weight_type = input_args[kInputIndex1]->GetType(); in InferType() local
Dbinary_cross_entropy_with_logits.cc35 auto weight_type = input_args[kInputIndex2]->GetType(); in InferType() local
Dbatch_norm_ext.cc112 auto weight_type = input_args[kInputIndex1]->GetType(); in InferType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dbce_with_logits_loss_cpu_kernel.cc73 auto weight_type = inputs[kIndex2]->GetType(); in Resize() local
168 auto weight_type = inputs[kIndex2]->GetType(); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_binary_cross_entropy.cc37 TypePtr weight_type; member
Dtest_ops_binary_cross_entropy_grad.cc39 TypePtr weight_type; member
Dtest_ops_binary_cross_entropy_with_logits.cc37 TypePtr weight_type; member
Dtest_ops_nllloss_grad.cc41 TypePtr weight_type; member
Dtest_ops_nllloss.cc42 TypePtr weight_type; member
Dtest_ops_batch_norm_ext.cc41 TypePtr weight_type; member
Dtest_ops_binary_cross_entropy_with_logits_backward.cc39 TypePtr weight_type; member
Dtest_ops_batch_norm_grad_ext.cc34 TypePtr weight_type; member