Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for TFE_TensorHandleCopyToDevice (0.15 sec)

  1. tensorflow/c/eager/c_api_cluster_test.cc

      TFE_TensorHandle* retvals[1];
      int num_retvals = 1;
      TFE_Execute(matmul, &retvals[0], &num_retvals, status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      auto* retval_task0 =
          TFE_TensorHandleCopyToDevice(retvals[0], ctx, local_device_name, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      CheckTFE_TensorHandleHasFloats(retval_task0, {7, 10, 15, 22});
    
      TFE_DeleteTensorHandle(retval_task0);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_experimental_test.cc

      for (int d = 0; d < num_devices; ++d) {
        const char* name = TF_DeviceListName(devices, d, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_TensorHandle* copy = TFE_TensorHandleCopyToDevice(m, ctx, name, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        void* data = TFE_TensorHandleDevicePointer(copy, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 03:14:26 UTC 2023
    - 31.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test.cc

        // Copy to device
        TFE_TensorHandle* hdevice =
            TFE_TensorHandleCopyToDevice(hcpu, ctx, name.c_str(), status.get());
        if (TF_GetCode(status.get()) != TF_OK) {
          ADD_FAILURE() << tag << " -- " << TF_Message(status.get());
          continue;
        }
        // Copy from device to the same device.
        TFE_TensorHandle* hdevice2 =
            TFE_TensorHandleCopyToDevice(hdevice, ctx, name.c_str(), status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_distributed_test.cc

      const char remote_device_name[] =
          "/job:localhost/replica:0/task:1/device:CPU:0";
      auto* h0_task1 =
          TFE_TensorHandleCopyToDevice(h0_task0, ctx, remote_device_name, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      auto* h1_task1 =
          TFE_TensorHandleCopyToDevice(h1_task0, ctx, remote_device_name, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    
      // Copying on to a parallel device must be explicit.
      TensorHandlePtr failed_copy_on_result(TFE_TensorHandleCopyToDevice(
          cpu_value.get(), context.get(), device_name, status.get()));
      EXPECT_EQ(TF_GetCode(status.get()), TF_UNIMPLEMENTED);
    
      std::array<TFE_TensorHandle*, 2> components{cpu_value.get(), cpu_value.get()};
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.h

    // If async execution is enabled, the copy may be enqueued and the call will
    // return "non-ready" handle. Else, this function returns after the copy has
    // been done.
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopyToDevice(
        TFE_TensorHandle* h, TFE_Context* ctx, const char* device_name,
        TF_Status* status);
    
    // Debugging/Profiling information for TFE_TensorHandle
    //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      std::vector<TensorHandlePtr> components;
      components.reserve(underlying_devices_.size());
      for (const std::string& underlying_device_name : underlying_devices_) {
        TFE_TensorHandle* t = TFE_TensorHandleCopyToDevice(
            tensor, context, underlying_device_name.c_str(), status);
        if (TF_GetCode(status) != TF_OK) return nullptr;
        components.emplace_back(t);
      }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api.cc

              unwrapped_op,
              reinterpret_cast<tensorflow::ImmediateExecutionTensorHandle**>(
                  retvals),
              num_retvals);
    }
    
    TFE_TensorHandle* TFE_TensorHandleCopyToDevice(TFE_TensorHandle* h,
                                                   TFE_Context* ctx,
                                                   const char* device_name,
    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