Function CopySliceToElement

tensorflow/core/util/batch_util.cc:162–181

Copies the index^th slice of parent (in the 0th dimension) into element.

// Copies the index^th slice of parent (in the 0th dimension) into element.
Status CopySliceToElement(const Tensor& parent, Tensor* element, int64 index) {
  TF_RETURN_IF_ERROR(ValidateInput(parent, *element, index));

#define HANDLE_TYPE(T)                               \
  case DataTypeToEnum<T>::value: {                   \
    HandleSliceToElement<T>(parent, element, index); \
    return Status::OK();                             \
  }

  switch (parent.dtype()) {
    TF_CALL_ALL_TYPES(HANDLE_TYPE);
    TF_CALL_QUANTIZED_TYPES(HANDLE_TYPE);
    TF_CALL_uint32(HANDLE_TYPE);
    TF_CALL_uint64(HANDLE_TYPE);
#undef HANDLE_TYPE
    default:
      return errors::Unimplemented("CopySliceToElement Unhandled data type: ",
                                   element->dtype());
  }
}

// Copies the index^th slice of parent (in the 0th dimension) into element.
//