Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for dev_dims (0.13 sec)

  1. tensorflow/c/eager/c_api_debug.cc

      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return nullptr;
      }
    
      std::vector<int64_t> dev_dims = TensorShapeAsVector(*handle, &status->status);
      if (!status->status.ok()) {
        return nullptr;
      }
      return new TFE_TensorDebugInfo(dev_dims);
    }
    
    TF_CAPI_EXPORT extern void TFE_DeleteTensorDebugInfo(
        TFE_TensorDebugInfo* debug_info) {
      delete debug_info;
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

      EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a));
      EXPECT_EQ(TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b));
    
      int64_t new_dims[] = {3, 2};
      TF_Status* status = TF_NewStatus();
      TF_TensorBitcastFrom(a, TF_UINT64, b, new_dims, 2, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status));
      TF_DeleteStatus(status);
    
      EXPECT_EQ(6, TF_TensorElementCount(a));
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top