Searched refs:labels_type (Results 1 – 2 of 2) sorted by relevance
| /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
2531 args = {"logits": logits_type, "labels": labels_type}
2599 def infer_dtype(self, logits_type, labels_type): argument
2602 … validator.check_tensor_dtype_valid("labels", labels_type, (mstype.int32, mstype.int64), self.name)
2954 def infer_dtype(self, acts_type, labels_type, input_length_type, label_length_type): argument
2959 (labels_type, input_length_type, label_length_type)))
|