Searched refs:ms_dtype (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_pyfunc_op.py | 47 def func_with_dtype(ms_dtype, np_dtype): argument
56 net = PyFuncNet(func_single_output, [ms_dtype, ms_dtype], [shape, shape], [ms_dtype], [shape])
129 ms_dtype = ms.int32
135 net = PyFuncGraph(func_single_output, [ms_dtype, ms_dtype], [shape, shape], [ms_dtype], [shape])
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 87 auto ms_dtype = AnfAlgo::GetOutputDeviceDataType(input_node, input_index); in ConvertInputShapeAndType() local
92 ge::DataType ge_dtype = ascend::GeTypesConvert::TransTypeIdToGeDataType(ms_dtype); in ConvertInputShapeAndType()
118 auto ms_dtype = AnfAlgo::GetOutputDeviceDataType(node, i); in ConvertOutputShapeAndType() local
122 ge::DataType ge_dtype = ascend::GeTypesConvert::TransTypeIdToGeDataType(ms_dtype); in ConvertOutputShapeAndType()
157 auto ms_dtype = AnfAlgo::GetPrevNodeOutputDeviceDataType(node, index); in NewConstantOp() local
166 ge::DataType ge_dtype = ascend::GeTypesConvert::TransTypeIdToGeDataType(ms_dtype); in NewConstantOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/ |
| D | trt_utils.h | 51 static std::variant<bool, nvinfer1::DataType> MsDtypeToTrtDtype(const TypeId &ms_dtype) { in MsDtypeToTrtDtype() argument
57 auto iter = type_list.find(ms_dtype); in MsDtypeToTrtDtype()
59 MS_LOG(WARNING) << "data type not support: " << ms_dtype; in MsDtypeToTrtDtype()
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/ |
| D | roi_align.py | 103 self.ms_dtype = mstype.float16 if _mode_16 else mstype.float32
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | math.py | 154 self.ms_dtype = mstype.float32
156 self.ms_dtype = mstype.int32
157 self.result_tensor = Tensor(data, dtype=self.ms_dtype)
|