Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for OnlyOutput (0.23 sec)

  1. tensorflow/c/experimental/ops/gen/cpp/renderers/op_implementation_renderer.cc

      ArgView output_arg = op_.OnlyOutput();
      Statement("int num_retvals = $0.size()", output_arg.VariableName());
      Statement("return " + Call(op_.VariableName(), "Execute",
                                 {output_arg.VariableName(), "&num_retvals"}));
    }
    
    void OpImplementationRenderer::RenderExecutionSingleOutput() {
      ArgView output_arg = op_.OnlyOutput();
      Statement("int num_retvals = 1");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 28 05:51:40 UTC 2024
    - 3.7K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/ops/gen/cpp/views/op_view.cc

    int OpView::NumOutputs() const { return output_args_.size(); }
    
    ArgView OpView::OnlyInput() const {
      CHECK_EQ(input_args_.size(), 1);  // Crash OK
      return input_args_.front();
    }
    
    ArgView OpView::OnlyOutput() const {
      CHECK_EQ(output_args_.size(), 1);  // Crash OK
      return output_args_.front();
    }
    
    string OpView::FunctionName() const { return op_.name(); }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/ops/gen/cpp/views/op_view.h

      const std::vector<AttrView> &Attributes() const;
      const std::vector<OpArgumentView> &AllArguments() const;
    
      int NumInputs() const;
      int NumOutputs() const;
      ArgView OnlyInput() const;
      ArgView OnlyOutput() const;
    
      string FunctionName() const;
      string VariableName() const;
      string OpNameString() const;
      string Summary() const;
      std::vector<string> Description() const;
      bool IsListOp() const;
    
     private:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 15 18:23:40 UTC 2021
    - 2K bytes
    - Viewed (0)
Back to top