Home
last modified time | relevance | path

Searched refs:CopySliceToElement (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dbatch_util.h33 Status CopySliceToElement(const Tensor& parent, Tensor* element, int64 index);
Dbatch_util.cc147 Status CopySliceToElement(const Tensor& parent, Tensor* element, int64 index) { in CopySliceToElement() function
/external/tensorflow/tensorflow/core/kernels/
Dqueue_base.cc337 Status QueueBase::CopySliceToElement(const Tensor& parent, Tensor* element, in CopySliceToElement() function in tensorflow::QueueBase
339 return batch_util::CopySliceToElement(parent, element, index); in CopySliceToElement()
Dqueue_base.h79 static Status CopySliceToElement(const Tensor& parent, Tensor* element,
Dfifo_queue.cc98 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
Dpriority_queue.cc126 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
Drandom_shuffle_queue_op.cc174 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op.cc142 batch_util::CopySliceToElement(t, &out_tensors->back(), i_)); in GetNextInternal()