Searched refs:ATTR_DTYPE (Results 1 – 4 of 4) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
D | tbe_json_creator.cc | 36 static std::unordered_map<std::string, ATTR_DTYPE> type_attr_dtype_map = { 37 {kVTypeInt, ATTR_DTYPE::ATTR_INT32}, 38 {kVTypeInt64, ATTR_DTYPE::ATTR_INT64}, 39 {kVTypeStr, ATTR_DTYPE::ATTR_STR}, 40 {kVTypeBool, ATTR_DTYPE::ATTR_BOOL}, 41 {kVTypeFloat, ATTR_DTYPE::ATTR_FLOAT32}, 42 {kVTypeListInt, ATTR_DTYPE::ATTR_LIST_INT32}, 43 {kVTypeListFloat, ATTR_DTYPE::ATTR_LIST_FLOAT32}, 44 {kVTypeListUInt64, ATTR_DTYPE::ATTR_LIST_UINT64}, 45 {kVTypeListListInt, ATTR_DTYPE::ATTR_LIST_LIST_INT64}}; [all …]
|
D | tbe_json_creator.h | 33 enum ATTR_DTYPE { enum
|
/third_party/mindspore/mindspore/core/utils/ |
D | flags.h | 45 inline const char ATTR_DTYPE[] = "dtype"; variable
|
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
D | prim.cc | 323 dtype_tuple[i] = out[ATTR_DTYPE]; in AbstractTupleToPython() 342 dic[ATTR_DTYPE] = dtype_tuple; in AbstractTupleToPython() 376 dtype_list[i] = out[ATTR_DTYPE]; in AbstractListToPython() 388 dic[ATTR_DTYPE] = dtype_list; in AbstractListToPython() 426 (*dic)[ATTR_DTYPE] = arg_tensor->BuildType(); in ConvertAbstractTensorToPython() 434 (*dic)[ATTR_DTYPE] = abs_base->BuildType(); in ConvertAbstractFunctionToPython() 442 (*dic)[ATTR_DTYPE] = std::make_shared<TypeType>(); in ConvertAbstractFunctionToPython() 458 dic[ATTR_DTYPE] = arg->BuildType(); in ConvertAbstractToPython() 463 dic[ATTR_DTYPE] = arg->BuildType(); in ConvertAbstractToPython() 468 dic[ATTR_DTYPE] = abs_base->BuildType(); in ConvertAbstractToPython() [all …]
|