Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for Donato (0.17 sec)

  1. tensorflow/c/eager/custom_device_test.cc

      EXPECT_TRUE(executed);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_DeleteTensorHandle(retval);
    
      // Explicit placement still forces the op onto the requested device
      matmul.reset(MatMulOp(context.get(), hcustom0.get(), hcpu.get()));
      TFE_OpSetDevice(matmul.get(), "/job:localhost/replica:0/task:0/device:CPU:0",
                      status.get());
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/custom_device_testutil.cc

    #include "tensorflow/core/platform/test.h"
    
    namespace {
    
    struct LoggingDevice {
      tensorflow::string device_name;
      tensorflow::string underlying_device;
      // Set to true whenever a TensorHandle is copied onto the device
      bool* arrived_flag;
      // Set to true whenever an operation is executed
      bool* executed_flag;
      // If true, only explicit op placements are accepted. If false, uses
      // type-based dispatch.
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TensorHandlePtr value_one(FloatTensorHandle(1., status.get()));
      TensorHandlePtr value_two(FloatTensorHandle(2., status.get()));
      {
        // Try to pack two TensorHandles onto a parallel device with a single
        // component.
        ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
        std::array<TFE_TensorHandle*, 2> components{value_one.get(),
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
Back to top