Home
last modified time | relevance | path

Searched refs:CopyElementToSlice (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dbatch_util.h30 Status CopyElementToSlice(Tensor element, Tensor* parent, int64 index);
Dbatch_util.cc172 Status CopyElementToSlice(Tensor element, Tensor* parent, int64 index) { in CopyElementToSlice() function
/external/tensorflow/tensorflow/core/kernels/
Dqueue_base.cc343 Status QueueBase::CopyElementToSlice(const Tensor& element, Tensor* parent, in CopyElementToSlice() function in tensorflow::QueueBase
345 return batch_util::CopyElementToSlice(element, parent, index); in CopyElementToSlice()
Dqueue_base.h85 static Status CopyElementToSlice(const Tensor& element, Tensor* parent,
Dmap_kernels.h213 OP_REQUIRES_OK(ctx, batch_util::CopyElementToSlice(it->first, result, i)); in Compute()
Dfifo_queue.cc332 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
Dpadding_fifo_queue.cc211 batch_util::CopyElementToSlice( in TryDequeueMany()
Dpriority_queue.cc363 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
Drandom_shuffle_queue_op.cc411 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
Dbarrier_ops.cc168 batch_util::CopyElementToSlice(std::move(ready_tuples[b][i]), in TryInsertMany()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsliding_window_dataset_op.cc232 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in GetNextInternal()
Dmap_and_batch_dataset_op.cc454 Status copy_status = batch_util::CopyElementToSlice( in CallFunction()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc156 return batch_util::CopyElementToSlice(val, out, iter_); in SetRetval()
Dpadded_batch_dataset_op.cc309 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in GetNextInternal()
Ddataset_utils.cc874 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in CopyBatch()
/external/tensorflow/tensorflow/core/framework/
Dtensor.h53 Status CopyElementToSlice(Tensor element, Tensor* parent, int64 index);
681 friend Status batch_util::CopyElementToSlice(