Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for ControlInputs (0.19 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
Back to top