tensorflow/c/eager/parallel_device/parallel_device.cc
// Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
// number of dimensions of a parallel tensor.
Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
number of dimensions of a parallel tensor.
Used as an argument to TFE_NewCustomDeviceTensorHandle for computing the number of dimensions of a parallel tensor .
int ParallelTensorNumDims(void* data, TF_Status* status) {
  const std::vector<int64_t>* shape;
  Status s = reinterpret_cast<ParallelTensor*>(data)->Shape(&shape);
  if (!s.ok()) {
    Set_TF_Status_from_Status(status, s);
    return -1;
  }
  return shape->size();
}
int ParallelTensorNumDims ( void * data , TF_Status * status ) { const std :: vector < int64_t > * shape ; Status s = reinterpret_cast < ParallelTensor * > ( data ) -> Shape ( & shape ) ; if ( ! s . ok ( ) ) { Set_TF_Status_from_Status ( status , s ) ; return -1 ; } return shape -> size ( ) ; }
Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
number of dimensions of a parallel tensor.
Used as an argument to TFE_NewCustomDeviceTensorHandle for computing the number of dimensions of a parallel tensor .