Searched refs:execute_op (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | remove_device_attribute.cc | 50 module.walk([&](tfrt::corert::ExecuteOp execute_op) { in runOnOperation() argument
54 execute_op.getOpFuncAttrs(&op_func_attrs); in runOnOperation()
55 execute_op.getOpAttrs(&op_attrs); in runOnOperation()
63 OpBuilder builder(execute_op); in runOnOperation()
65 execute_op.getLoc(), execute_op.getResultTypes(), in runOnOperation()
66 execute_op.op_handler(), execute_op.operands(), new_op_attrs, in runOnOperation()
67 op_func_attrs, execute_op.op_name()); in runOnOperation()
68 execute_op.replaceAllUsesWith(new_execute_op.results()); in runOnOperation()
69 execute_op.erase(); in runOnOperation()
|
| D | cross_device_transfer.cc | 80 } else if (auto execute_op = llvm::dyn_cast<tfrt::corert::ExecuteOp>(op)) { in GetDevice() local
82 execute_op.getOpAttrs(&attrs); in GetDevice()
|
| D | tf_to_tfrt.cc | 1840 if (auto execute_op = in GetNumArgs() local
1842 return execute_op.operands().size(); in GetNumArgs()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | tpu_add_resource_device_attribute.cc | 52 mlir::BlockArgument arg, mlir::TF::TPUExecuteOp execute_op) { in AddPlaceholderDeviceAttributeToResource() argument
55 execute_op->getParentOfType<mlir::tf_device::LaunchOp>(); in AddPlaceholderDeviceAttributeToResource()
58 auto function = execute_op->getParentOfType<mlir::func::FuncOp>(); in AddPlaceholderDeviceAttributeToResource()
59 mlir::OpBuilder builder(execute_op); in AddPlaceholderDeviceAttributeToResource()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 440 OpBuilder* builder, TF::TPUExecuteOp* execute_op) { in BuildExecuteOp() argument
451 *execute_op = builder->create<TF::TPUExecuteOp>(cluster_func.getLoc(), in BuildExecuteOp()
455 (*execute_op)->setAttr("_producer_name", producer_name_attr); in BuildExecuteOp()
589 Operation* execute_op, OpBuilder* builder) { in AssignDevicesToReplicatedExecute() argument
596 return WrapOpInLaunch(builder, execute_op->getLoc(), execute_op, device); in AssignDevicesToReplicatedExecute()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.h | 113 const ChainedExecuteFn& execute_op,
|
| D | xrt_util.cc | 382 const ChainedExecuteFn& execute_op, in ExecuteChained() argument
410 execute_op(op, working_set.PinnedTuples())); in ExecuteChained()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | critical_section_test.py | 160 execute_op = [
165 execute_op,
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 430 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local
470 config, execute_op, backend->memory_allocator()); in DoWork()
|
| D | xrt_execute_op.cc | 562 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local
578 device_ref.device_ordinal(), plan, config, execute_op, in DoWork()
|