Searched refs:element_access (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fifo_queue.cc | 94 Tensor* element_access = nullptr; in GetElementComponentFromBatch() local
96 tuple[component].dtype(), element_shape, out_tensor, &element_access)); in GetElementComponentFromBatch()
98 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
|
| D | padding_fifo_queue.cc | 62 Tensor* element_access = nullptr; in GetElementComponent() local
64 tuple[component].dtype(), element_shape, out_tensor, &element_access)); in GetElementComponent()
65 *element_access = tuple[component]; in GetElementComponent()
|
| D | priority_queue.cc | 122 Tensor* element_access = nullptr; in GetElementComponentFromBatch() local
124 tuple[component].dtype(), element_shape, out_tensor, &element_access)); in GetElementComponentFromBatch()
126 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
|
| D | random_shuffle_queue_op.cc | 171 Tensor* element_access = nullptr; in GetElementComponentFromBatch() local
173 tuple[component].dtype(), element_shape, out_tensor, &element_access)); in GetElementComponentFromBatch()
175 batch_util::CopySliceToElement(tuple[component], element_access, index)); in GetElementComponentFromBatch()
|