- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for replaceAllUsesInRegionWith (0.32 sec)
-
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
new_args.reserve(extern_values.size()); Block& block = func_region.front(); for (Value value : extern_values) { auto arg = block.addArgument(value.getType(), loc); replaceAllUsesInRegionWith(value, arg, func_region); new_args.push_back(arg); } // Replace yield op with return. Operation* yield_op = outlined_func.getBody().front().getTerminator(); OpBuilder b(yield_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
} // Add constant at start of region. auto const_builder = OpBuilder::atBlockBegin(®ion->front()); auto const_value = const_builder.clone(*extern_value.getDefiningOp()); replaceAllUsesInRegionWith(extern_value, const_value->getResult(0), *region); } } return llvm::to_vector<4>(extern_values); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
for (Operation* op : topological_order) op->moveBefore(block, block->end()); // Replace the 'inputs' values with the new function's arguments. for (auto p : llvm::zip(inputs, new_func.getArguments())) replaceAllUsesInRegionWith(std::get<0>(p), std::get<1>(p), new_func.getBody()); builder.setInsertionPointToEnd(block); builder.create<func::ReturnOp>(parent_func.getLoc(), outputs.getArrayRef());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
builder, cluster, /*before=*/true, launch_block, host_device); for (auto result : llvm::zip(launch.GetBody().getTerminator()->getOperands(), launch.getResults())) replaceAllUsesInRegionWith(std::get<0>(result), std::get<1>(result), cluster.getBody()); } // Extracts and move outside compiled ops that have no dependencies in the // cluster to before the cluster.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
replicate_op.GetBody().getArguments())) { mlir::TF::TPUReplicatedInputOp input = std::get<0>(input_and_block_arg); Value block_arg = std::get<1>(input_and_block_arg); mlir::replaceAllUsesInRegionWith(input->getResult(0), block_arg, cluster.getBody()); // Update replicated input use in tf.TPUPartitionedInputV2 op. for (auto& use : input->getUses()) { auto pi =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0)