- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 81 for GetBody (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
rewriter.inlineRegionBefore(graph.getNodes(), executor_graph.getBody(), executor_graph.getBody().end()); // Add terminator of tf_executor::graph rewriter.setInsertionPointToEnd(&executor_graph.getBody().front()); rewriter.create<tf_executor::FetchOp>(loc); // Add terminator of func rewriter.setInsertionPointToEnd(&func.getBody().front()); rewriter.create<func::ReturnOp>(loc);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/uniform_quantized_types_test.cc
auto func_op = module_op->lookupSymbol<func::FuncOp>("fully_quantized_add"); ASSERT_THAT(func_op, NotNull()); auto add_op_itr = func_op.getBody().op_begin<mlir::stablehlo::AddOp>(); ASSERT_THAT(add_op_itr, Ne(func_op.getBody().op_end<mlir::stablehlo::AddOp>())); EXPECT_TRUE(IsOpFullyQuantized(*add_op_itr)); } TEST_F(IsOpFullyQuantizedTest, FalseIfOpNotQuantized) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
/*operands=*/llvm::ArrayRef<Value>{}, cluster->getAttrs()); new_cluster.getBody().takeBody(cluster.getBody()); auto operand_not_in_cluster = [&](mlir::OpOperand& operand) { return !new_cluster.getOperation()->isProperAncestor(operand.getOwner()); }; for (auto result : llvm::zip(new_cluster.GetBody().getTerminator()->getOperands(), new_cluster.getResults()))
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/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// and packed args. arg.replaceAllUsesWith( new_replicate.GetBody().getArgument(arg.getArgNumber() + 1)); } } for (auto& op : llvm::make_early_inc_range(replicate.GetBody())) { op.moveBefore(&new_replicate.GetBody(), new_replicate.GetBody().end()); } replicate.replaceAllUsesWith(new_replicate); replicate.erase(); return new_replicate; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
IRMapping& mapper) { Block& main_body = dst_graph_op.GetBody(); // Take the reference of the main graph's FetchOp to later move to the end. FetchOp main_fetch_op = dst_graph_op.GetFetch(); Block& save_func_body = src_graph_op.GetBody(); for (Operation& op : save_func_body.without_terminator()) { main_body.push_back(op.clone(mapper)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K 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/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/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)