- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for GetRegionOutputs (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
results_to_remove_by_region.resize(num_regions); for (int i = 0; i < num_regions; ++i) { Block& block = parallel_execute.GetRegionBlockWithIndex(i); auto results = parallel_execute.GetRegionOutputs(i); auto& results_to_remove = results_to_remove_by_region[i]; results_to_remove.reserve(results.size()); Operation* terminator = block.getTerminator(); for (auto result : llvm::enumerate(results)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_input_output_aliases.cc
if (auto execute = llvm::dyn_cast_or_null<tf_device::ParallelExecuteOp>(parent)) { int operand_idx = use->getOperandNumber(); auto execute_results = execute.GetRegionOutputs( device_return->getParentRegion()->getRegionNumber()); if (operand_idx >= execute_results.size()) return nullptr; auto result_from_use = execute_results[operand_idx];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 04:14:26 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
®ion_output_index); if (failed(status)) return mlir::failure(); const auto output_from_logical_device = new_parallel_execute.GetRegionOutputs(cluster_idx + core_id)[region_output_index]; outputs_to_merge->emplace_back(output_from_logical_device); } return mlir::success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
llvm::SmallVector<bool, 8> output_merged(execute_launch.getNumResults(), false); auto execute_outputs = parallel_execute ? parallel_execute.GetRegionOutputs( execute_launch->getParentRegion()->getRegionNumber()) : execute_launch.getResults(); for (auto execute_output : llvm::enumerate(execute_outputs)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
); let regions = (region VariadicRegion<SizedRegion<1>>:$regions); let extraClassDeclaration = [{ Block& GetRegionBlockWithIndex(unsigned index); Operation::result_range GetRegionOutputs(unsigned region_index); // Checks if a tf_device.parallel_execute index'th region block wraps a // single operation and the single operation results are perfectly forwarded // to the region block's return.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
state.addTypes(output_types); } Block& ParallelExecuteOp::GetRegionBlockWithIndex(unsigned index) { return getOperation()->getRegion(index).front(); } Operation::result_range ParallelExecuteOp::GetRegionOutputs( unsigned region_index) { int num_region_results = GetRegionBlockWithIndex(region_index).getTerminator()->getNumOperands(); int return_value_offset = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0)