Searched refs:GetInputOriShapeForTbeBuild (Results 1 – 3 of 3) sorted by relevance
153 auto ori_shape = TbeJsonUtils::GetInputOriShapeForTbeBuild(anf_node, real_input_index); in GenInputDescJson()353 auto ori_shape = TbeJsonUtils::GetInputOriShapeForTbeBuild(anf_node, real_input_index); in GenInputDescJson()406 auto ori_shape = TbeJsonUtils::GetInputOriShapeForTbeBuild(anf_node, real_input_index); in GenInputDescJson()
116 …static std::vector<int64_t> GetInputOriShapeForTbeBuild(const AnfNodePtr &anf_node, size_t real_id…
79 std::vector<int64_t> TbeJsonUtils::GetInputOriShapeForTbeBuild(const AnfNodePtr &anf_node, size_t r… in GetInputOriShapeForTbeBuild() function in mindspore::kernel::TbeJsonUtils