Searched refs:batch_util (Results 1 – 21 of 21) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | batch_util_test.cc | 27 auto s = batch_util::CopyContiguousSlices( in TEST()
35 auto s = batch_util::CopyContiguousSlices( in TEST()
43 auto s = batch_util::CopyContiguousSlices( in TEST()
52 auto s = batch_util::CopyContiguousSlices( in TEST()
|
| D | tensor.h | 52 namespace batch_util {
681 friend Status batch_util::CopyElementToSlice(
684 friend Status batch_util::CopySliceToElement(
687 friend Status batch_util::MaybeMoveSliceToElement(
690 friend Status batch_util::CopyContiguousSlices(
|
| /external/tensorflow/tensorflow/core/util/ |
| D | batch_util.h | 22 namespace batch_util {
|
| D | BUILD | 77 "batch_util.cc",
78 "batch_util.h",
142 "batch_util.h",
210 "batch_util.cc",
332 "batch_util.h",
|
| D | batch_util.cc | 25 namespace batch_util { namespace
|
| /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()
363 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 175 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
411 attempt->context->SetStatus(batch_util::CopyElementToSlice( in TryDequeueMany()
|
| D | map_kernels.h | 213 OP_REQUIRES_OK(ctx, batch_util::CopyElementToSlice(it->first, result, i)); in Compute()
|
| 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 | 5723 "batch_util.h",
7223 # tensorflow/core/util/batch_util.h
7225 name = "batch_util",
7226 hdrs = ["batch_util.h"],
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 295 TF_RETURN_IF_ERROR(batch_util::SetElementZero( in GetNextInternal()
309 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in GetNextInternal()
313 TF_RETURN_IF_ERROR(batch_util::CopyElementToLargerSlice( in GetNextInternal()
|
| D | tensor_slice_dataset_op.cc | 141 batch_util::CopySliceToElement(t, &out_tensors->back(), index)); in GetNextInternal()
|
| D | map_defun_op.cc | 156 return batch_util::CopyElementToSlice(val, out, iter_); in SetRetval()
|
| D | dataset_utils.cc | 874 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in CopyBatch()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unbatch_dataset_op.cc | 137 TF_RETURN_IF_ERROR(batch_util::MaybeMoveSliceToElement( in GetNextInternal()
|
| D | sliding_window_dataset_op.cc | 232 TF_RETURN_IF_ERROR(batch_util::CopyElementToSlice( in GetNextInternal()
|
| D | rebatch_dataset_op.cc | 510 TF_RETURN_IF_ERROR(batch_util::CopyContiguousSlices( in GetNextInternal()
|
| D | map_and_batch_dataset_op.cc | 454 Status copy_status = batch_util::CopyElementToSlice( in CallFunction()
|