Searched defs:device_func (Results 1 – 6 of 6) sorted by relevance
101 FuncOp device_func, in AddAliasingAttributeToDeviceFunc()136 FuncOp device_func = module.lookupSymbol<FuncOp>(func_attr.getValue()); in runOnOperation() local
617 auto device_func = cluster_func->getFunc(); in runOnOperation() local
595 def device_func(op): function
1169 def device_func(op): function
718 auto* device_func = SparseApplyFtrlKernel<T, Tindex, has_l2_shrinkage>; in operator ()() local
3253 def device_func(unused_op): function