Home
last modified time | relevance | path

Searched refs:execute_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc464 Operation* execute_op = BuildExecuteOp(compile_op, launch_func, builder); in Rewrite() local
465 launch_func.replaceAllUsesWith(execute_op); in Rewrite()
477 execute_op->setAttr(kDeviceAttr, in Rewrite()
Dtpu_variable_runtime_reformatting.cc376 for (auto execute_op : in HandleReplicateOp() local
379 execute = execute_op; in HandleReplicateOp()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.h74 const ChainedExecuteFn& execute_op);
Dxrt_util.cc240 const ChainedExecuteFn& execute_op) { in ExecuteChained() argument
267 execute_op(op, working_set.PinnedTuples())); in ExecuteChained()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcritical_section_test.py164 execute_op = [
169 execute_op,
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc371 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local
386 device_ref.device_ordinal(), plan, config, execute_op); in DoWork()