Home
last modified time | relevance | path

Searched defs:logits_dtype (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_sparseSoftmaxCrossEntropyWithLogits.py36 def np_sparse_softmax_cross_entropy_with_logits(labels_shape, logits_shape, logits_dtype): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Drandom_categorical.cc109 auto logits_dtype = logits->GetDataType(); in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Drandom_ops_proto.cc272 DataType logits_dtype = logits_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
307 DataType logits_dtype = logits_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Drandom_categorical_cpu_kernel.cc31 #define ADD_KERNEL(logits_dtype, nun_sample_dtype, seed_dtype, output_dtype, logits_type, output_ty… argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/loss/
Dloss.py2334 logits_dtype, label_dtype, argument