- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for SinkResourceWritesIntoParallelExecute (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
} // Finds AssignVariableOps that can be moved into a parallel_execute region and // moves them. Leftover parallel_execute results that were used by the // such AssignVariableOp are also pruned. void SinkResourceWritesIntoParallelExecute( tf_device::ParallelExecuteOp parallel_execute) { bool rewrite = false; const int num_regions = parallel_execute.getNumRegions(); llvm::SmallVector<Value, 4> results_to_remap;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0)