Searched defs:ms_dtype (Results 1 – 3 of 3) sorted by relevance
87 auto ms_dtype = AnfAlgo::GetOutputDeviceDataType(input_node, input_index); in ConvertInputShapeAndType() local118 auto ms_dtype = AnfAlgo::GetOutputDeviceDataType(node, i); in ConvertOutputShapeAndType() local157 auto ms_dtype = AnfAlgo::GetPrevNodeOutputDeviceDataType(node, index); in NewConstantOp() local
51 static std::variant<bool, nvinfer1::DataType> MsDtypeToTrtDtype(const TypeId &ms_dtype) { in MsDtypeToTrtDtype()
47 def func_with_dtype(ms_dtype, np_dtype): argument