Function CopyElementToSlice

tensorflow/core/util/batch_util.cc:138–159

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

// Copies element into the index^th slice of parent (in the 0th dimension).
Status CopyElementToSlice(Tensor element, Tensor* parent, int64 index) {
  TF_RETURN_IF_ERROR(ValidateInput(*parent, element, index));
  const int64 num_values = element.NumElements();
  bool can_move = element.RefCountIsOne();
#define HANDLE_TYPE(T)                                               \
  case DataTypeToEnum<T>::value: {                                   \
    T* src = element.base<T>();                                      \
    T* dest = parent->base<T>() + (num_values * index);              \
    return HandleElementToSlice<T>(src, dest, num_values, can_move); \
  }

  switch (element.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("CopyElementToSlice Unhandled data type: ",
                                   element.dtype());
  }
}

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