Searched refs:batch_util (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | batch_util.h | 22 namespace batch_util {
|
| D | batch_util.cc | 25 namespace batch_util { namespace
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 324 TF_RETURN_IF_ERROR(batch_util::SetElementZero( in GetNextInternal()
336 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in GetNextInternal()
339 TF_RETURN_IF_ERROR(batch_util::CopyElementToLargerSlice( in GetNextInternal()
|
| D | tensor_slice_dataset_op.cc | 142 batch_util::CopySliceToElement(t, &out_tensors->back(), i_)); in GetNextInternal()
|
| D | batch_dataset_op.cc | 205 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in GetNextInternal()
|
| D | map_defun_op.cc | 241 return batch_util::CopyElementToSlice(val, out, iter_); in SetRetval()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | queue_base.cc | 339 return batch_util::CopySliceToElement(parent, element, index); in CopySliceToElement()
345 return batch_util::CopyElementToSlice(element, parent, index); in CopyElementToSlice()
|
| D | fifo_queue.cc | 98 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
332 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
|
| D | priority_queue.cc | 126 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
362 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 174 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
410 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
|
| D | padding_fifo_queue.cc | 211 batch_util::CopyElementToSlice( in TryDequeueMany()
|
| D | barrier_ops.cc | 168 batch_util::CopyElementToSlice(std::move(ready_tuples[b][i]), in TryInsertMany()
|
| D | BUILD | 5417 "batch_util.h",
7094 # tensorflow/core/util/batch_util.h
7096 name = "batch_util",
7097 hdrs = ["batch_util.h"],
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unbatch_dataset_op.cc | 113 TF_RETURN_IF_ERROR(batch_util::MaybeMoveSliceToElement( in GetNextInternal()
|
| D | sliding_window_dataset_op.cc | 223 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in GetNextInternal()
|
| D | map_and_batch_dataset_op.cc | 469 Status copy_status = batch_util::CopyElementToSlice( in CallFunction()
|
| D | numa_map_and_batch_dataset_op.cc | 1104 s.Update(batch_util::CopyElementToSlice(std::move(tensor), in WorkerThread()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 50 namespace batch_util {
629 friend Status batch_util::CopyElementToSlice(
632 friend Status batch_util::MaybeMoveSliceToElement(
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 943 "util/batch_util.h",
|