Searched defs:host_compute (Results 1 – 2 of 2) sorted by relevance
245 auto host_compute = builder.create<TF::_XlaHostComputeMlirOp>( in CreateHostCompute() local292 auto host_compute = CreateHostCompute( in MoveOpsToHost() local
327 TF::_XlaHostComputeMlirOp host_compute, in RewriteHostComputeOp()771 if (auto host_compute = dyn_cast<TF::_XlaHostComputeMlirOp>(curr_op)) { in RewriteFunction() local