Home
last modified time | relevance | path

Searched defs:host_compute (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_extract_outside_compilation.cc406 auto host_compute = builder.create<TF::_XlaHostComputeMlirOp>( in CreateHostCompute() local
476 TF::_XlaHostComputeMlirOp host_compute) { in ReplaceExternalOutputUsage()
538 auto host_compute = in MoveOpsToHost() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_communication.cc324 TF::_XlaHostComputeMlirOp host_compute, in RewriteHostComputeOp()
837 if (auto host_compute = dyn_cast<TF::_XlaHostComputeMlirOp>(curr_op)) { in RewriteFunction() local