- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for GetRegionBlockWithIndex (0.82 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/parallel_execute_util.cc
auto parallel_execute = builder->create<tf_device::ParallelExecuteOp>( cluster_func.getLoc(), 1, output_types); cluster_func->remove(); auto& block = parallel_execute.GetRegionBlockWithIndex(0); builder->setInsertionPointToEnd(&block); builder->insert(cluster_func); cluster_func.replaceAllUsesWith(parallel_execute); builder->create<tf_device::ReturnOp>(block.getParent()->getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
llvm::SmallVector<llvm::SmallVector<int, 4>, 4> results_to_remove_by_region; 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();
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/utils/xla_rewrite_util.cc
// will be replaced. int cluster_idx = -1; for (size_t child_idx = 0; child_idx < old_parallel_execute.getRegions().size(); ++child_idx) { auto& block = old_parallel_execute.GetRegionBlockWithIndex(child_idx); if (cluster_func->getBlock() == &block) { assert(cluster_idx == -1); cluster_idx = child_idx; } } assert(cluster_idx != -1); // Move children that are preserved.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 4.4K 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) -
tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc
const int num_regions = parallel_execute_op.getOperation()->getNumRegions(); executes.reserve(num_regions); for (int i : llvm::seq<int>(0, num_regions)) { Block& execute_block = parallel_execute_op.GetRegionBlockWithIndex(i); // Replace terminator with tf_executor.YieldOp. Operation* terminator = execute_block.getTerminator(); builder->setInsertionPoint(terminator);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 19 19:47:16 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
}]; let results = (outs Variadic<AnyType>:$execute_outputs ); 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
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/tf2xla/internal/passes/extract_outside_compilation.cc
SmallVector<mlir::tf_device::LaunchOp, 4> core_to_host_launch; for (int core = 0; core < core_to_tmp_host_launch.size(); ++core) { Block& host_computation_block = parallel_execute_op.GetRegionBlockWithIndex(core); builder.setInsertionPointToEnd(&host_computation_block); // map_outside_compilation with return values from host is not implemented.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
int end) { const int size_before = output_types->size(); for (int index = start; index < end; ++index) { Block& block = parallel_execute.GetRegionBlockWithIndex(index); auto terminator_operand_types = block.getTerminator()->getOperandTypes(); output_types->append(terminator_operand_types.begin(), terminator_operand_types.end()); }
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/transforms/host_runtime/tpu_rewrite_pass.cc
assert(input_list.size() == num_cores_per_replica); for (int core = 0; core < num_cores_per_replica; ++core) { auto& block = new_parallel_execute->GetRegionBlockWithIndex((*cluster_idx) + core); builder->setInsertionPointToEnd(&block); // Create Execute op. // // TODO(b/148913294): Identify inputs/return values specific to each
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)