Searched defs:creater_type (Results 1 – 3 of 3) sorted by relevance
98 …explicit TbeKernelJsonCreator(kCreaterType creater_type = SINGLE_BUILD) : creater_type_(creater_ty… in creater_type_() argument
257 mindspore::kernel::kCreaterType creater_type) { in GenTopKV2IndicesTensorInfo()
149 std::vector<int64_t> GetOutputDeviceShapeForTbeBuild(const kCreaterType creater_type, const AnfNode… in GetOutputDeviceShapeForTbeBuild()171 std::vector<int64_t> GetInputDeviceShapeForTbeBuild(const kCreaterType creater_type, const AnfNodeP… in GetInputDeviceShapeForTbeBuild()