Home
last modified time | relevance | path

Searched refs:ATTR_DTYPE (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dtbe_json_creator.cc36 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 …]
Dtbe_json_creator.h33 enum ATTR_DTYPE { enum
/third_party/mindspore/mindspore/core/utils/
Dflags.h45 inline const char ATTR_DTYPE[] = "dtype"; variable
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprim.cc323 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 …]