Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for AddInput (0.09 sec)

  1. tensorflow/c/eager/gradients.cc

      forward_op_->op_name = op;
      forward_op_->attrs.Reset(op);
      return op_->Reset(op, raw_device_name);
    }
    absl::Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input,
                          ForwardOperation* forward_op_) {
      TF_RETURN_IF_ERROR(op_->AddInput(input));
      forward_op_->inputs.push_back(input);
      return absl::OkStatus();
    }
    absl::Status AddInputList(AbstractOperation* op_,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients_internal.h

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

      if (isa<TracingOperation>(check_numerics_op.get())) {
        s = dyn_cast<TracingOperation>(check_numerics_op.get())
                ->SetOpName("check_numerics");
        ASSERT_EQ(errors::OK, s.code()) << s.message();
      }
      s = AddInput(check_numerics_op.get(), t.get(), &forward_op);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
      string message = "This is the way!";
      s = SetAttrString(check_numerics_op.get(), "message", message.data(),
    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

      //
      // The value will override the previous value - that is, no "merging" of
      // 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,
    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.cc

                             TF_AbstractTensor* const* inputs, TF_OutputList* o,
                             TF_Status* s) {
      for (int i = 0; i < num_inputs; i++) {
        tsl::Set_TF_Status_from_Status(s, unwrap(op)->AddInput(unwrap(inputs[i])));
        if (TF_GetCode(s) != TF_OK) {
          return;
        }
      }
      int num_outputs = unwrap(o)->expected_num_outputs;
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Execute(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      absl::Status SetDeviceName(const char* name) override {
        // TODO(srbs): Implement this.
        device_name_ = name;
        return absl::OkStatus();
      }
    
      absl::Status AddInput(AbstractTensorHandle* input) override {
        GraphTensor* t = dyn_cast<GraphTensor>(input);
        if (!t) {
          return tensorflow::errors::InvalidArgument(
              "Unable to cast input to GraphTensor");
        }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api.cc

      return tensorflow::unwrap(op)->DeviceName().c_str();
    }
    
    void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) {
      status->status = tensorflow::unwrap(op)->AddInput(tensorflow::unwrap(input));
    }
    
    void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs,
                            TF_Status* status) {
      status->status = tensorflow::unwrap(op)->AddInputList(
    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