- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for ControlInputs (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td
If any operand or implicitly captured value are dead, the region is not executed and dead values are immediately returned for every result. }]; let arguments = (ins Variadic<TfeControlType>:$controlInputs ); let results = (outs Variadic<AnyType>:$outputs, TfeControlType:$control ); let regions = (region SizedRegion<1>:$body); let extraClassDeclaration = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 23 19:35:12 UTC 2023 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
auto wrapper_island_op = builder.create<IslandOp>( loc, /*outputs=*/TypeRange{}, /*control=*/tf_executor::ControlType::get(builder.getContext()), /*controlInputs=*/control_dependencies); wrapper_island_op.getBody().emplaceBlock(); // Create a NoOp inside the IslandOp. auto guard = OpBuilder::InsertionGuard(builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/cc/framework/scope.cc
} void Scope::UpdateBuilder(NodeBuilder* builder) const { std::vector<Node*> control_inputs; for (const auto& op : impl()->control_deps_) { control_inputs.push_back(op.node()); } builder->ControlInputs(control_inputs); if (!impl()->kernel_label_.empty()) { builder->Attr("_kernel", impl()->kernel_label_); } if (!impl()->colocation_constraints_.empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
op->getLoc(), op->getOperand(0).getType()); builder.create<tf_executor::NextIterationSinkOp>( op->getLoc(), source_op.getToken(), /*input=*/op->getOperand(0), /*controlInputs=*/new_operands); op->replaceAllUsesWith( ValueRange({source_op.getOutput(), source_op.getControl()})); op->erase(); }); } class ConvertGraphOp : public OpConversionPattern<tfg::GraphOp> {
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/tensorflow/transforms/executor_island_coarsening.cc
data_types.push_back(value.getType()); } } auto island = OpBuilder(fetch).create<IslandOp>( fetch.getLoc(), data_types, /*control=*/ControlType::get(fetch.getContext()), /*controlInputs=*/control_fetches); island.getBody().push_back(new Block); OpBuilder::atBlockEnd(&island.GetBody()) .create<YieldOp>(fetch.getLoc(), data_fetches); const int fetch_control_idx = data_fetches.size();
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/executor_tpuv1_island_coarsening.cc
} } IslandOp new_island = OpBuilder(island).create<IslandOp>( island.getLoc(), result_types, /*control=*/ControlType::get(island.getContext()), /*controlInputs=*/island.getOperands()); new_island.getBody().push_back(new Block); // Move the operations in the new island, gather the results of the new yield. Block& island_body = new_island.GetBody();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis_test.cc
EXPECT_NE(predicate_map[ControlOutputFor(next_iterations[0])], ""); EXPECT_NE(predicate_map[ControlOutputFor(next_iterations[1])], ""); } } TEST(DeadnessAnalysisTest, ControlInputs) { Scope root = Scope::NewRootScope().ExitOnError(); ops::Switch sw = CreateSwitch(root, "0"); Output id0 = ops::Identity(root.WithOpName("id0"), sw.output_false);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 51.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
given by the control dependencies. }]; // May take values of type !tfl.control as inputs let arguments = (ins Variadic<TFL_Control>:$controlInputs ); // Adds a single value of type !tfl.control as the last output (all other // outputs are copied from the wrapped region.) let results = (outs Variadic<AnyTensor>:$outputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)