Searched defs:GetInputDeviceShapeForTbeBuild (Results 1 – 2 of 2) sorted by relevance
85 std::vector<int64_t> TbeJsonUtils::GetInputDeviceShapeForTbeBuild(const AnfNodePtr &anf_node, size_… in GetInputDeviceShapeForTbeBuild() function in mindspore::kernel::TbeJsonUtils
171 std::vector<int64_t> GetInputDeviceShapeForTbeBuild(const kCreaterType creater_type, const AnfNodeP… in GetInputDeviceShapeForTbeBuild() function