Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for AddInput (0.23 sec)

  1. tensorflow/c/c_test_util.cc

      TF_AddInput(desc, l);
      TF_AddInput(desc, r);
      return TF_FinishOperation(desc, s);
    }
    
    TF_Operation* RandomUniform(TF_Operation* shape, TF_DataType dtype,
                                TF_Graph* graph, TF_Status* s) {
      TF_OperationDescription* desc =
          TF_NewOperation(graph, "RandomUniform", "random_uniform");
      TF_AddInput(desc, {shape, 0});
      TF_SetAttrType(desc, "dtype", dtype);
    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)
  2. tensorflow/c/eager/gradients.cc

      forward_op_->op_name = op;
      forward_op_->attrs.Reset(op);
      return op_->Reset(op, raw_device_name);
    }
    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();
    }
    Status AddInputList(AbstractOperation* op_,
    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)
  3. tensorflow/c/c_api_test.cc

        if (transpose_a) {
          TF_SetAttrBool(desc, "transpose_a", 1);
        }
        if (transpose_b) {
          TF_SetAttrBool(desc, "transpose_b", 1);
        }
        TF_AddInput(desc, {l, 0});
        TF_AddInput(desc, {r, 0});
        TF_Operation* op = TF_FinishOperation(desc, s);
        EXPECT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        return op;
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  4. tensorflow/c/eager/c_api_experimental_test.cc

      ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
      TF_OperationDescription* id_descr =
          TF_NewOperation(function_graph, "Identity", "id");
      TF_SetAttrType(id_descr, "T", TF_INT32);
      TF_AddInput(id_descr, {arg, 0});
      TF_Operation* id = TF_FinishOperation(id_descr, status);
      ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
      TF_Output input{arg, 0};
      TF_Output output{id, 0};
      TF_Function* fn =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  5. 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 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,
    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)
  6. tensorflow/c/eager/gradients_internal.h

    // Records the op name in the `ForwardOperation`.
    Status Reset(AbstractOperation*, const char* op, const char* raw_device_name,
                 ForwardOperation*);
    
    // Records the inputs in the `ForwardOperation`.
    Status AddInput(AbstractOperation*, AbstractTensorHandle*, ForwardOperation*);
    Status AddInputList(AbstractOperation*,
                        absl::Span<AbstractTensorHandle* const> inputs,
                        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)
  7. 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(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        device_name_ = name;
        return absl::OkStatus();
      }
    
      Status AddInput(AbstractTensorHandle* input) override {
        GraphTensor* t = dyn_cast<GraphTensor>(input);
        if (!t) {
          return tensorflow::errors::InvalidArgument(
              "Unable to cast input to GraphTensor");
        }
        TF_AddInput(op_.get(), t->output_);
        return absl::OkStatus();
      }
    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)
  9. tensorflow/c/eager/unified_api_testutil.cc

        }
    
        AbstractOperationPtr fn_op(ctx->CreateOperation());
        TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr));
        for (auto input : inputs) {
          TF_RETURN_IF_ERROR(fn_op->AddInput(input));
        }
        int retvals = outputs.size() - null_indices.size();
        std::vector<AbstractTensorHandle*> fn_outputs(retvals);
        TF_RETURN_IF_ERROR(fn_op->Execute(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  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
    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