Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TFE_TensorHandleCopySharingTensor (0.26 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device.cc

                "placed on the parallel device)."));
            TF_SetStatus(status, TF_INVALID_ARGUMENT, message.c_str());
            return result;
          }
          components.emplace_back(TFE_TensorHandleCopySharingTensor(
              absl::get<TFE_TensorHandle*>(inputs[i]), status));
        }
        std::vector<MaybeParallelTensorOwned> result_content;
        result_content.reserve(1);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.h

    // with `h`. On success, `status` is set to OK. On failure, `status` reflects
    // the error and a nullptr is returned.
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor(
        TFE_TensorHandle* h, TF_Status* status);
    
    // This function will block till the operation that produces `h` has
    // completed. The memory returned might alias the internal memory used by
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
  3. tensorflow/c/eager/c_api.cc

        return nullptr;
      }
      return tensorflow::unwrap(h)->BackingDeviceName(&status->status);
    }
    
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor(
        TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
    
    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)
  4. tensorflow/c/eager/c_api_test.cc

      TFE_DeleteContextOptions(opts);
    
      TFE_TensorHandle* h = TestMatrixTensorHandle(ctx);
      EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h));
    
      TFE_TensorHandle* h_shares_tensor =
          TFE_TensorHandleCopySharingTensor(h, status.get());
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      TF_Tensor* t = TFE_TensorHandleResolve(h_shares_tensor, status.get());
      ASSERT_EQ(16, TF_TensorByteSize(t));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
Back to top