Searched defs:data_dtype (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_min.py | 34 def min_max_case(op_func, np_func, data_dtype=np.float32, has_nan=False, save_graphs=False): argument
81 def min_max_case_all_dyn(op_func, data_dtype=np.float32): argument
106 def test_min(mode, data_dtype): argument
150 def test_min_all_dynamic(data_dtype): argument
166 def test_min_tensor(mode, data_dtype): argument
185 def test_min_control_flow(mode, data_dtype): argument
|
| D | test_ops_max.py | 36 def test_max(mode, data_dtype): argument
80 def test_max_all_dynamic(data_dtype): argument
96 def test_min_tensor(mode, data_dtype): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/ |
| D | error_util.cc | 70 const std::string &data_dtype) { in GetInputDtypeNotSupportErrMsg()
77 const std::string &data_dtype) { in GetInputDTypeErrMsg()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/custom_ops_tbe/ |
| D | cus_square.py | 36 def infer_dtype(self, data_dtype): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | dynamic_stitch_cpu_kernel.cc | 82 auto data_dtype = inputs[kIndex1]->dtype_id(); in UpdateOutputShapeAndSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/ |
| D | _constexpr_utils.py | 496 def check_tensors_dtype_same(data_dtype, value_dtype, op_name): argument
556 def convert_scalar_to_tensor(data_shape, data_dtype, indices_shape, value, op_type): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _thor_ops.py | 336 def infer_dtype(self, data_dtype): argument
|
| D | comm_ops.py | 90 def check_collective_target_dtype(data_name, data_dtype, prim_name): argument
|