- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 187 for GetBody (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
// arguments. Region& op_region = op.getBody(); for (auto p : llvm::zip(live_ins, outlined_func_block->getArguments())) { replaceAllUsesInRegionWith(std::get<0>(p), std::get<1>(p), op_region); } // Move all instructions in cluster_op into outlined_function's only block. auto& op_body = op.GetBody().getOperations(); outlined_func_block->getOperations().splice(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
removed_params.set(i); callee.getBody().addArgument(param.getType(), param.getLoc()); param.replaceAllUsesWith(callee.getArguments().back()); removed_params.push_back(false); } } // Remove old resource-type parameters. callee.getBody().front().eraseArguments(removed_params); // Update function type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
// Map block arg to replica arg. mapping.clear(); for (auto& block_arg : replicate_op.GetBody().getArguments()) mapping.map(block_arg, replicate_op.GetReplicaOperandForBlockArgument(block_arg, i)); // Copy over replicate region into replica island. replicate_op.getBody().cloneInto(&replica.getBody(), mapping); if (failed(UpdateRegionReplicateVariantOps(builder, replicate_op.getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc
llvm::SmallVectorImpl<Operation*>* ops_to_visit) { if (auto island = llvm::dyn_cast<IslandOp>(op)) { mlir::visitUsedValuesDefinedAbove( island.getBody(), island.getBody(), [&](OpOperand* operand) { VisitOpOperand(graph, operand->get(), reachable_ops, ops_to_visit); }); } VisitOpOperands(graph, op, reachable_ops, ops_to_visit);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
type = FunctionType::get(context, types, result_types); } auto outlined_func = builder.create<func::FuncOp>(loc, name, type); outlined_func.getBody().takeBody(region); Region& func_region = outlined_func.getBody(); // Replace all external uses with block args and update uses. llvm::SmallVector<Value, 4> new_args; new_args.reserve(extern_values.size());
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/quantization/tensorflow/passes/merge_duplicate_resource_ops.cc
// one resource op and one Yield op. auto island_op = llvm::dyn_cast_or_null<IslandOp>(op); if (!island_op || !island_op.getBody().hasOneBlock()) return nullptr; auto& island_block = island_op.getBody().front(); if (++island_block.begin() != --island_block.end()) return nullptr; Operation* resource_op = &island_block.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 04:26:16 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc
island_op.getLoc(), llvm::ArrayRef<Type>{}, island_op.getControl().getType(), island_operands); island_sink.getBody().push_back(new Block); builder.setInsertionPointToEnd(&island_sink.GetBody()); builder.create<tf_executor::YieldOp>(island_op.getLoc(), llvm::ArrayRef<Value>{});
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/transforms/executor_island_coarsening.cc
// Add associated inner op result to operands of the YieldOp. yield_operands.push_back(old_result.inner_op_result); } // Create YieldOp for the new island. OpBuilder builder(&new_island.GetBody(), new_island.GetBody().end()); return builder.create<YieldOp>(new_island.getLoc(), yield_operands); } // Moves inner ops (excluding last op/YieldOp) from islands being merged into // the new merged island.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
for (Value value : graph_op.GetFetch().getFetches()) { graph_fetches.insert(value); } for (Operation& op : graph_op.GetBody().without_terminator()) { auto island = dyn_cast<tf_executor::IslandOp>(&op); if (!island) continue; if (!island.WrapsSingleOp()) continue; Operation& wrapped_op = island.GetBody().front(); if (!IsStatefulFunctionalControlFlowOp(&wrapped_op)) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
platforms/core-execution/workers/src/testFixtures/groovy/org/gradle/workers/fixtures/WorkerExecutorFixture.groovy
} abstract String getBody() } class WorkParameterClass extends TestClass { Map<String, String> fields = [:] WorkParameterClass(String name, String packageName) { super(name, packageName) this.imports += ["org.gradle.workers.WorkParameters"] } String getBody() { return """
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Fri Jan 05 12:36:12 UTC 2024 - 14.2K bytes - Viewed (0)