Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_TensorHandleNumDims (0.13 sec)

  1. tensorflow/c/eager/c_api_distributed_test.cc

          TFE_CreatePackedTensorHandle(ctx, handles.data(), &num_replicas, status);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      EXPECT_EQ(TFE_TensorHandleDataType(packed_handle), TF_RESOURCE);
      EXPECT_EQ(TFE_TensorHandleNumDims(packed_handle, status), 0);
      EXPECT_EQ(TFE_TensorHandleNumElements(packed_handle, status), 1);
    
      const string composite_device_name =
          "/job:localhost/replica:0/task:0/device:COMPOSITE:0";
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TensorHandlePtr combined_value = CreatePerDeviceValues(
          context.get(), components, device_name, status.get());
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
      int num_axes = TFE_TensorHandleNumDims(combined_value.get(), status.get());
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
      EXPECT_EQ(num_axes, 1);
    }
    
    TEST(PARALLEL_DEVICE, TestNestedParallelDevices) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

      if (h) {
        tensorflow::unwrap(h)->Unref();
      }
    }
    
    TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) {
      return static_cast<TF_DataType>(tensorflow::unwrap(h)->DataType());
    }
    
    int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return -1;
      }
    
      int num_dims = -1;
    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