Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CopyTensorHandleToDevice (0.16 sec)

  1. tensorflow/c/eager/immediate_execution_context.h

      virtual ImmediateExecutionTensorHandle* CreateLocalHandle(
          AbstractTensorInterface* t) = 0;
      // Copy the handle to another device.
      virtual ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
          ImmediateExecutionTensorHandle* handle, const char* device_name,
          absl::Status* status) = 0;
    
      // Create an operation to perform op execution
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

        return nullptr;
      }
    
      tensorflow::ImmediateExecutionContext* unwrapped_ctx =
          tensorflow::unwrap(ctx);
    
      auto* result =
          unwrapped_ctx->GetCustomDeviceOpHandler().CopyTensorHandleToDevice(
              unwrapped_ctx, tensorflow::unwrap(h), device_name, &status->status);
    
      if (status->status.ok()) {
        return tensorflow::wrap(result);
      }
      return nullptr;
    }
    
    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