Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for AddInputList (0.11 sec)

  1. tensorflow/c/eager/gradients_internal.h

    // Records the inputs in the `ForwardOperation`.
    absl::Status AddInput(AbstractOperation*, AbstractTensorHandle*,
                          ForwardOperation*);
    absl::Status AddInputList(AbstractOperation*,
                              absl::Span<AbstractTensorHandle* const> inputs,
                              ForwardOperation*);
    
    // Sets the attrs in the `ForwardOperation`.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

      forward_op_->inputs.push_back(input);
      return absl::OkStatus();
    }
    absl::Status AddInputList(AbstractOperation* op_,
                              absl::Span<AbstractTensorHandle* const> inputs,
                              ForwardOperation* forward_op_) {
      TF_RETURN_IF_ERROR(op_->AddInputList(inputs));
      for (auto input : inputs) {
        forward_op_->inputs.push_back(input);
      }
      return absl::OkStatus();
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients_test.cc

          "or NotDifferentiableGradientFunction.",
          s.message());
      ASSERT_EQ(nullptr, outputs[0]);
    }
    
    // TODO(b/164171226): Enable this test with tfrt after AddInputList is
    // supported. It is needed for IdentityN.
    #ifdef PLATFORM_GOOGLE
    INSTANTIATE_TEST_SUITE_P(
        UnifiedCAPI, CppGradients,
        ::testing::Combine(::testing::Values("graphdef", "mlir"),
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/abstract_operation.h

      // existing and given constraints will be performed.
      virtual absl::Status SetDeviceName(const char* name) = 0;
    
      virtual absl::Status AddInput(AbstractTensorHandle* input) = 0;
      virtual absl::Status AddInputList(
          absl::Span<AbstractTensorHandle* const> inputs) = 0;
      virtual absl::Status Execute(absl::Span<AbstractTensorHandle*> retvals,
                                   int* num_retvals) = 0;
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        if (!t) {
          return tensorflow::errors::InvalidArgument(
              "Unable to cast input to GraphTensor");
        }
        TF_AddInput(op_.get(), t->output_);
        return absl::OkStatus();
      }
      absl::Status AddInputList(
          absl::Span<AbstractTensorHandle* const> inputs) override {
        std::vector<TF_Output> tf_outputs(inputs.size());
        for (int i = 0; i < inputs.size(); i++) {
          GraphTensor* t = dyn_cast<GraphTensor>(inputs[i]);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.cc

    }
    
    void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs,
                            TF_Status* status) {
      status->status = tensorflow::unwrap(op)->AddInputList(
          {reinterpret_cast<tensorflow::AbstractTensorHandle**>(
               tensorflow::unwrap(inputs)),
           static_cast<size_t>(num_inputs)});
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top