Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for pick (0.23 sec)

  1. tensorflow/c/eager/c_api.cc

            TFE_DeleteTensorHandle(outputs[i]);
          }
        }
        return status.status;
      }
    
      tensorflow::Status Pack(absl::Span<ImmediateExecutionTensorHandle*> handles,
                              ImmediateExecutionTensorHandle** result) override {
        TF_Status status;
        *result = tensorflow::unwrap(device_.pack(context_,
                                                  tensorflow::wrap(handles.data()),
    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)
  2. tensorflow/c/eager/c_api_experimental.cc

          // One of the inputs we're trying to pack is on a custom device. We'll let
          // the first custom device we see handle all of the packing.
          auto* custom_device_handle =
              tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>(
                  unwrapped_handle);
          tensorflow::ImmediateExecutionTensorHandle* result;
          status->status = custom_device_handle->device()->Pack(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
Back to top