Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_outside_compilation.cc406 auto host_compute = builder.create<TF::_XlaHostComputeMlirOp>( in CreateHostCompute() local
477 TF::_XlaHostComputeMlirOp host_compute) { in ReplaceExternalOutputUsage()
539 auto host_compute = in MoveOpsToHost() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc334 TF::_XlaHostComputeMlirOp host_compute, in RewriteHostComputeOp()
781 if (auto host_compute = dyn_cast<TF::_XlaHostComputeMlirOp>(curr_op)) { in RewriteFunction() local