Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 13 for AddInputList (0.35 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.h

      const string& Name() const override;
      const string& DeviceName() const override;
      Status SetDeviceName(const char* name) override;
      Status AddInput(AbstractTensorHandle* input) override;
      Status AddInputList(absl::Span<AbstractTensorHandle* const> inputs) override;
      Status Execute(absl::Span<AbstractTensorHandle*> retvals,
                     int* num_retvals) override;
      Status SetAttrString(const char* attr_name, const char* data,
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 23 23:12:39 GMT 2020
    - 3.7K bytes
    - Viewed (1)
  2. tensorflow/c/experimental/gradients/tape/tape_operation.cc

      TF_RETURN_IF_ERROR(parent_op_->AddInput(input));
      forward_op_.inputs.push_back(input);
      return OkStatus();
    }
    Status TapeOperation::AddInputList(
        absl::Span<AbstractTensorHandle* const> inputs) {
      TF_RETURN_IF_ERROR(parent_op_->AddInputList(inputs));
      for (auto input : inputs) {
        forward_op_.inputs.push_back(input);
      }
      return OkStatus();
    }
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  3. tensorflow/c/eager/gradients_internal.h

                 ForwardOperation*);
    
    // Records the inputs in the `ForwardOperation`.
    Status AddInput(AbstractOperation*, AbstractTensorHandle*, ForwardOperation*);
    Status AddInputList(AbstractOperation*,
                        absl::Span<AbstractTensorHandle* const> inputs,
                        ForwardOperation*);
    
    // Sets the attrs in the `ForwardOperation`.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:35 GMT 2021
    - 4.2K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.cc

      TF_RETURN_IF_ERROR(op_->AddInput(input));
      forward_op_->inputs.push_back(input);
      return absl::OkStatus();
    }
    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();
    }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_test_util.cc

                     TF_Status* s, const char* name, TF_Operation** op,
                     bool check) {
      TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name);
      TF_Output add_inputs[2] = {{l, 0}, {r, 0}};
      TF_AddInputList(desc, add_inputs, 2);
      *op = TF_FinishOperation(desc, s);
      if (check) {
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        ASSERT_NE(*op, nullptr);
      }
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  6. 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"),
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  7. 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();
      }
      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]);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  8. tensorflow/c/eager/abstract_operation.h

      // existing and given constraints will be performed.
      virtual Status SetDeviceName(const char* name) = 0;
    
      virtual Status AddInput(AbstractTensorHandle* input) = 0;
      virtual Status AddInputList(
          absl::Span<AbstractTensorHandle* const> inputs) = 0;
      virtual Status Execute(absl::Span<AbstractTensorHandle*> retvals,
                             int* num_retvals) = 0;
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
  9. tensorflow/c/c_api.h

    // single tensor), and TF_AddInputList() for the second input (since
    // it takes a list, even if you were to pass a list with a single
    // tensor), as in:
    //   TF_OperationDescription* desc = TF_NewOperation(graph, "Concat", "c");
    //   TF_Output concat_dim_input = {...};
    //   TF_AddInput(desc, concat_dim_input);
    //   TF_Output values_inputs[5] = {{...}, ..., {...}};
    //   TF_AddInputList(desc, values_inputs, 5);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  10. tensorflow/c/eager/c_api.h

    // TFE_DeleteOp() is called before TFE_DeleteContext().
    //
    // Very similar to TF_OperationDescription with some differences:
    // (1) TF_Output or TFE_TensorHandle* as arguments to TF_AddInput,
    //     TF_AddInputList
    // (2) TF_ColocateWith, TF_AddControlInput etc. do not make sense.
    // (3) Implementation detail: Avoid use of NodeBuilder/NodeDefBuilder since
    //     the additional sanity checks there seem unnecessary;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
Back to top