Searched refs:kTBE (Results 1 – 5 of 5) sorted by relevance
71 case kTBE: in ImplTypeToStr()88 OpImplyType imply_type = kTBE; in RegOp()210 if (imply_type == kTBE) { in DecodeOpInfo()266 if (imply_type == kTBE) { in DecodeAttr()326 if (imply_type == kTBE) { in DecodeInputOutput()360 if (is_gpu && (imply_type == kTBE || imply_type == kAICPU)) { in FindOp()
28 enum OpImplyType { kAKG = 0, kTBE = 1, kAICPU = 2, kCPU }; enumerator174 OpImplyType imply_type_ = kTBE;
371 return mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kTBE, is_dynamic_shape); in FindOp()
345 if (kernel::OpLib::FindOp(prim::kPrimCast->name(), kernel::kTBE) != nullptr) { in AddCastOpNodeToGraph()
177 auto op_info = mindspore::kernel::OpLib::FindOp(op_name, kernel::kTBE); in HasRefNodes()