Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for TFE_OpAddInput (0.19 sec)

  1. tensorflow/c/eager/c_api_test_util.cc

      TF_Status* status = TF_NewStatus();
    
      TFE_Op* op = TFE_NewOp(ctx, "AddV2", status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddInput(op, a, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddInput(op, b, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TF_DeleteStatus(status);
      TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a));
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  2. tensorflow/c/eager/c_api_test.cc

      int num_retvals = 1;
      for (auto s : state) {
        TFE_OpReset(matmul, "MatMul", nullptr, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_OpAddInput(matmul, m, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_OpAddInput(matmul, m, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_Execute(matmul, &retvals[0], &num_retvals, status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  3. tensorflow/c/eager/c_api_cluster_test.cc

      TF_Status* status = TF_NewStatus();
      TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpAddInput(op, var, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      int num_retvals = 1;
      TFE_Execute(op, out_value, &num_retvals, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_distributed_test.cc

      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
    
      TFE_Op* func = TFE_NewOp(ctx, "AddVariablesFunction", status);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      TFE_OpAddInput(func, packed_handle, status);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      if (remote) {
        TFE_OpSetDevice(func, task1_name, status);
        ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental_test.cc

        ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
        TF_DeleteTensor(t);
    
        TFE_Op* op = TFE_NewOp(ctx, "ident", status);
        ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
        TFE_OpAddInput(op, h, status);
        ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
    
        std::vector<TFE_TensorHandle*> result;
        result.push_back(nullptr);
        int num_retvals = 1;
    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)
  6. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern const char* TFE_OpGetDevice(const TFE_Op* op,
                                                      TF_Status* status);
    
    TF_CAPI_EXPORT extern void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input,
                                              TF_Status* status);
    
    TF_CAPI_EXPORT extern void TFE_OpAddInputList(TFE_Op* op,
    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)
  7. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        if (TF_GetCode(status) != TF_OK) return;
      }
      TFE_OpAddAttrs(op_.get(), attributes);
      for (int input_index = 0; input_index < inputs.size(); ++input_index) {
        TFE_OpAddInput(op_.get(), inputs[input_index], status);
        if (TF_GetCode(status) != TF_OK) return;
      }
      std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs);
      int real_num_outputs = expected_max_outputs;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  8. tensorflow/c/eager/c_api.cc

      status->status = tensorflow::unwrap(op)->SetDeviceName(device_name);
    }
    
    const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status) {
      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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
Back to top