Searched defs:data_dtype (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/ops/custom_ops_tbe/ |
| D | cus_square.py | 36 def infer_dtype(self, data_dtype): argument
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | _constexpr_utils.py | 433 def check_tensors_dtype_same(data_dtype, value_dtype, op_name): argument
473 def convert_scalar_to_tensor(data_shape, data_dtype, indices_shape, value, op_type): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _thor_ops.py | 336 def infer_dtype(self, data_dtype): argument
|
| D | _quant_ops.py | 1772 def infer_dtype(self, data_dtype, data_min_dtype, data_max_dtype, cumsum_dtype): argument
|