Searched refs:in_op_chain (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_kernels.cc | 136 void BatchFunctionFallback(tfrt::Argument<Chain> in_op_chain, in BatchFunctionFallback() argument
179 auto op_chain = in_op_chain.ValueRef(); in BatchFunctionFallback()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| D | tfrt_fallback_async.cc | 288 p << "(" << op.in_op_chain() << ") key(" in print()
314 p << "(" << op.in_op_chain() << ", " << op.allocator() << ") key(" in print()
326 p << "(" << in_op_chain() << ") " << getOperation()->getAttr("f") << " (" in print()
|
| D | tfrt_fallback_async.td | 394 ExecuteOp. In addition, it takes an in_op_chain and produces an out_op_chain
396 in_op_chain.
420 TFRT_ChainType:$in_op_chain,
465 TFRT_ChainType:$in_op_chain,
518 TFRT_ChainType:$in_op_chain,
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_async.td | 394 ExecuteOp. In addition, it takes an in_op_chain and produces an out_op_chain
396 in_op_chain.
420 TFRT_ChainType:$in_op_chain,
466 TFRT_ChainType:$in_op_chain,
|
| D | tfrt_fallback_async.cc | 220 p << "tfrt_fallback_async.executeop.seq(" << op.in_op_chain() << ") key(" in print()
232 p << "tfrt_fallback_async.batch_function(" << op.in_op_chain() << ") " in print()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | remote_run_encapsulate.cc | 147 loc, chain_type, remote_exec_op.in_op_chain(), remote_exec_op.context(), in runOnOperation()
|
| D | tf_to_tfrt.cc | 491 mlir::Value in_op_chain = in matchAndRewrite() local
534 in_op_chain, distributed_context, task_handle, op.callee(), in matchAndRewrite()
|