Method Dim

tensorflow/core/common_runtime/eager/tensor_handle.cc:301–305

}

Status TensorHandle::Dim(int dim_index, int64* dim) {
  DCHECK(dim != nullptr);
  TF_RETURN_IF_ERROR(WaitReady());
  return tensor_handle_data_->Dim(dim_index, dim);
}

Status TensorHandle::NumElements(int64* num_elements) {
  DCHECK(num_elements != nullptr);