Searched defs:device_func (Results 1 – 6 of 6) sorted by relevance
101 func::FuncOp device_func, in AddAliasingAttributeToDeviceFunc()136 func::FuncOp device_func = in runOnOperation() local
618 auto device_func = cluster_func->getFunc(); in runOnOperation() local
595 def device_func(op): function
1165 def device_func(op): function
718 auto* device_func = SparseApplyFtrlKernel<T, Tindex, has_l2_shrinkage>; in operator ()() local
3305 def device_func(unused_op): function