Searched defs:execute_launch (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_dynamic_layout_pass.cc | 157 TF::TPUCopyWithLayoutOp BuildCopyWithLayout(tf_device::LaunchOp execute_launch, in BuildCopyWithLayout()
168 TF::TPUExecuteOp execute, tf_device::LaunchOp execute_launch, in HandleInput()
184 tf_device::LaunchOp execute_launch, tf_device::LaunchOp compile_launch, in HandleReplicatedInputs()
237 auto& execute_launch = std::get<0>(execute_and_input_mapping); in HandleCompileAndExecutes() local
296 auto execute_launch = in runOnFunction() local
|
| D | tpu_merge_variables_with_execute.cc | 126 tf_device::LaunchOp execute_launch, bool check_device, in BuildVariableAccessInfo()
332 tf_device::LaunchOp execute_launch, in ReplaceParallelExecute()
407 void ReplaceExecute(tf_device::LaunchOp execute_launch, in ReplaceExecute()
424 TF::_TPUCompileMlirOp GetTPUCompileOp(tf_device::LaunchOp execute_launch) { in GetTPUCompileOp()
437 LogicalResult UpdateSerializedModule(tf_device::LaunchOp execute_launch, in UpdateSerializedModule()
486 LogicalResult MergeForOneTPUExecute(tf_device::LaunchOp execute_launch, in MergeForOneTPUExecute()
607 for (auto execute_launch : execute_launches) { in runOnFunction() local
|
| D | tpu_colocate_composite_resource_ops.cc | 118 for (auto execute_launch : execute_launches) { in runOnFunction() local
|
| D | tpu_variable_runtime_reformatting.cc | 386 tf_device::LaunchOp execute_launch; in HandleReplicateOp() local
|