tensorflow/c/eager/parallel_device/parallel_device.cc
// Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing a
// dimension of a parallel tensor.
Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing a
dimension of a parallel tensor.
Used as an argument to TFE_NewCustomDeviceTensorHandle for computing a dimension of a parallel tensor .
int64_t ParallelTensorDim(void* data, int dim_index, 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)[dim_index];
}
int64_t ParallelTensorDim ( void * data , int dim_index , 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 ) [ dim_index ] ; }
Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing a
dimension of a parallel tensor.
Used as an argument to TFE_NewCustomDeviceTensorHandle for computing a dimension of a parallel tensor .