Home
last modified time | relevance | path

Searched refs:new_parallel_execute (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_parallel_execute_sink_resource_write.cc129 auto new_parallel_execute = builder.create<tf_device::ParallelExecuteOp>( in SinkResourceWritesIntoParallelExecute() local
132 for (auto region : llvm::zip(new_parallel_execute.getRegions(), in SinkResourceWritesIntoParallelExecute()
137 llvm::zip(results_to_remap, new_parallel_execute.getResults())) in SinkResourceWritesIntoParallelExecute()
Dtpu_merge_variables_with_execute.cc355 auto new_parallel_execute = builder->create<tf_device::ParallelExecuteOp>( in ReplaceParallelExecute() local
360 Operation* new_parallel_execute_op = new_parallel_execute.getOperation(); in ReplaceParallelExecute()