Home
last modified time | relevance | path

Searched refs:kJOriShape (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dsingle_tbe_json_creator.cc168 (*input_desc)[kJOriShape] = ori_shape; in GenInputDescJson()
344 (*output_desc)[kJOriShape] = ori_shape; in GenDescJson()
363 (*input_desc)[kJOriShape] = ori_shape; in GenInputDescJson()
397 (*output_desc)[kJOriShape] = ori_shape; in GenDescJson()
416 (*input_desc)[kJOriShape] = ori_shape; in GenInputDescJson()
Dtbe_json_utils.h39 constexpr auto kJOriShape = "ori_shape"; variable
Dtbe_json_creator.cc422 (*output_desc)[kJOriShape] = ori_shape; in GenDescJson()
547 if (output_desc.find(kJOriShape) != output_desc.end()) { in GenOutputDataDescJson()
Dfusion_tbe_json_creator.cc142 if (input_desc.find(kJOriShape) != input_desc.end()) { in GenDataJson()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc59 constexpr auto kJOriShape = "ori_shape"; variable
305 input_desc_json[kJOriShape] = ori_shape; in GenValidInputDescJson()
532 output_obj[kJOriShape] = ori_shape; in GenOutputList()
1118 (*output_desc)[kJOriShape] = ori_shape; in GenDescJson()