logits_type (definition) in projects: third_party

Searched defs:logits_type (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
D	softmax_cross_entropy_with_logits.cc	`47 auto logits_type = CheckAndConvertUtils::CheckTensorTypeSame(args, valid_types, prim_name); in SoftmaxCrossEntropyWithLogitsInfer() local`
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pipeline/infer/
D	primitive_test.py	`56 def infer_dtype(self, logits_type, labels_type): argument`
/third_party/mindspore/mindspore/ops/operations/
D	nn_ops.py	`2530 def infer_dtype(self, logits_type, labels_type): argument 2599 def infer_dtype(self, logits_type, labels_type): argument`