Searched refs:BatchT (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.h | 114 using BatchT = Batch<BatchResourceBase::BatchTask>; variable
160 static Status ValidateBatch(const BatchT& batch);
167 Status ConcatInputTensors(const BatchT& batch, OpKernelContext* context,
191 BatchT* batch) const;
193 void ProcessFuncBatch(std::unique_ptr<BatchT> batch) const;
196 void ProcessBatch(std::unique_ptr<BatchT> batch) const;
205 static Status EmitIndexTensor(OpKernelContext* context, const BatchT& batch,
|
| D | batch_resource_base.cc | 286 /*static*/ Status BatchResourceBase::ValidateBatch(const BatchT& batch) { in ValidateBatch()
318 const BatchT& batch, OpKernelContext* context, in ConcatInputTensors()
478 const std::vector<Tensor>& combined_outputs, BatchT* batch) const { in SplitOutputTensors()
550 void BatchResourceBase::ProcessFuncBatch(std::unique_ptr<BatchT> batch) const { in ProcessFuncBatch()
636 void BatchResourceBase::ProcessBatch(std::unique_ptr<BatchT> batch) const { in ProcessBatch()
706 const BatchT& batch, in EmitIndexTensor()
737 auto process_batch_callback = [this](std::unique_ptr<BatchT> batch) { in LookupOrCreateBatcherQueue()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_functor.cc | 90 using BatchT = typename std::conditional<B2S, const T, T>::type; typedef
96 typename TTypes<BatchT, NUM_BLOCK_DIMS + 2>::Tensor batch_tensor) { in operator ()()
|
| D | spacetobatch_functor_gpu.cu.cc | 104 using BatchT = typename std::conditional<B2S, const T, T>::type; typedef
110 typename TTypes<BatchT, NUM_BLOCK_DIMS + 2>::Tensor batch_tensor) { in operator ()()
|