Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 46 for TensorHandle (0.21 sec)

  1. tensorflow/c/eager/custom_device_test.cc

                            &arrived, &executed, status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_TensorHandle* hcpu = TestMatrixTensorHandle(context);
      ASSERT_FALSE(arrived);
      TFE_TensorHandle* hdevice =
          TFE_TensorHandleCopyToDevice(hcpu, context, name, status.get());
      ASSERT_TRUE(arrived);
      ASSERT_FALSE(executed);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test_util.cc

    using tensorflow::string;
    using tensorflow::tstring;
    
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, float value) {
      float data[] = {value};
      TF_Status* status = TF_NewStatus();
      TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status);
      memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t));
      TFE_TensorHandle* th = TFE_NewTensorHandleFromTensor(ctx, t, status);
    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)
  3. tensorflow/c/eager/c_api_experimental.h

    // returned will be on the device in which the TFE_TensorHandle resides (so e.g.
    // for a GPU tensor this will return a pointer to GPU memory). The pointer is
    // only guaranteed to be valid until TFE_DeleteTensorHandle is called on this
    // TensorHandle. Only supports POD data types.
    TF_CAPI_EXPORT extern void* TFE_TensorHandleDevicePointer(TFE_TensorHandle*,
                                                              TF_Status*);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_debug_test.cc

      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_Context* ctx = TFE_NewContext(opts, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_TensorHandle* h = TestScalarTensorHandle(ctx, 1.0f);
      TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 06 22:10:09 GMT 2020
    - 2.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/dlpack.cc

    // Gets tensor from eager tensor handle.
    const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      tensorflow::TensorHandle* handle =
          tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h));
      if (handle->Type() != TensorHandle::LOCAL) {
        status->status = tensorflow::errors::InvalidArgument(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.cc

    }
    
    TFE_TensorHandle* TFE_NewTensorHandle(const TF_Tensor* t, TF_Status* status) {
      tensorflow::Tensor tensor;
      status->status = tensorflow::TF_TensorToTensor(t, &tensor);
      if (!status->status.ok()) return nullptr;
    
      return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor));
    }
    
    void TFE_DeleteTensorHandle(TFE_TensorHandle* h) {
      if (h == nullptr) return;
    
    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)
  7. tensorflow/c/eager/c_api_remote_test_util.cc

      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
      EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
    
      TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx);
      TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx);
      std::vector<TFE_TensorHandle*> handles_task0;
      if (heavy_load_on_streaming_rpc) {
        // Send 50 tensor copy requests to simulate that there have been some RPC
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Dec 11 22:56:03 GMT 2020
    - 9.1K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_cluster_test.cc

      const char dev1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0";
    
      TFE_TensorHandle* var_handle0 = TestVariable(ctx, 1.0, dev0_name);
      EXPECT_NE(var_handle0, nullptr);
      TFE_TensorHandle* var_handle1 = TestVariable(ctx, 2.0, dev1_name);
      EXPECT_NE(var_handle1, nullptr);
    
      TFE_TensorHandle* value_handle = nullptr;
      ReadVariable(ctx, var_handle1, &value_handle);
    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)
  9. tensorflow/c/eager/c_api_unified_experimental_eager.cc

      }
      return wrap(static_cast<AbstractContext*>(unwrap(c_ctx)));
    }
    
    TF_AbstractTensor* TF_CreateAbstractTensorFromEagerTensor(TFE_TensorHandle* t,
                                                              TF_Status* s) {
      return wrap(static_cast<AbstractTensorHandle*>(unwrap(t)));
    }
    
    TFE_TensorHandle* TF_AbstractTensorGetEagerTensor(TF_AbstractTensor* at,
                                                      TF_Status* s) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jun 25 04:40:46 GMT 2020
    - 3.2K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_remote_test.cc

      TFE_DeleteContextOptions(opts);
    
      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx);
      TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx);
      const char remote_device_name[] =
          "/job:localhost/replica:0/task:1/device:CPU:0";
      auto* h0_task1 =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 12 00:14:22 GMT 2020
    - 5.4K bytes
    - Viewed (0)
Back to top