Home
last modified time | relevance | path

Searched defs:batches (Results 1 – 25 of 116) sorted by relevance

12345

/external/igt-gpu-tools/tests/i915/
Dgem_exec_gttfill.c38 struct batch *batches = array; in xchg_batch() local
49 struct batch *batches, unsigned int count) in submit()
113 struct batch *batches; in fillgtt() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h48 const int batches = output_shape.Dims(0); in FullyConnectedPerChannel() local
89 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnectedPerChannel() local
132 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
174 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
Dpooling.h34 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
97 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
152 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
215 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h47 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnectedSparseWeight() local
92 const int batches = thread_end - thread_start; in FullyConnectedSparseWeight1x16Impl() local
130 const int batches = thread_end - thread_start; in FullyConnectedSparseWeight1x4Impl() local
209 const int batches = in FullyConnectedSparseWeight1x16() local
234 const int batches = in FullyConnectedSparseWeight1x4() local
/external/pytorch/aten/src/ATen/benchmarks/
Dquantize_per_channel.cpp7 const size_t batches = static_cast<size_t>(state.range(0)); in quantize_per_channel_4d_contiguous() local
25 const size_t batches = static_cast<size_t>(state.range(0)); in quantize_per_channel_4d_channels_last() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpooling.h34 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
92 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
147 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in L2Pool() local
201 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
255 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
Dfully_connected.h44 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnected() local
89 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
136 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
189 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected() local
Dresize_bilinear.h66 int32_t batches = MatchingDim(input_shape, 0, output_shape, 0); in ResizeBilinear() local
154 const int32_t batches = MatchingDim(input_shape, 0, output_shape, 0); in ResizeBilinearInteger() local
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_unpooling_2d.cc37 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxUnpooling() local
85 int batches = input->dims->data[0]; in Prepare() local
/external/sdk-platform-java/gax-java/gax/src/main/java/com/google/api/gax/batching/
DAccumulatingBatchReceiver.java39 private final ConcurrentLinkedQueue<T> batches = new ConcurrentLinkedQueue<>(); field in AccumulatingBatchReceiver
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h53 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnectedPerChannel() local
125 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
/external/tensorflow/tensorflow/core/kernels/image/
Dadjust_contrast_op_benchmark_test.cc24 static Graph* AdjustContrast(int batches, int width, int height) { in AdjustContrast()
Dresize_op_benchmark_test.cc24 static Graph* Resize(const char* algorithm, int batches, int width, in Resize()
Dmirror_pad_op_benchmark_test.cc24 static Graph* MirrorPad(int batches, int height, int width, int depth, int pad, in MirrorPad()
Dnon_max_suppression_op_benchmark_test.cc24 static Graph* CombinedNonMaxSuppression(int batches, int box_num, int class_num, in CombinedNonMaxSuppression()
/external/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_batch_cache.h37 struct fd_batch *batches[32]; member
/external/ComputeLibrary/tests/validation/reference/
DIm2Col.cpp51 const int batches = src.shape().total_size_upper(3); in im2col_nchw() local
105 const int batches = src.shape().total_size_upper(3); in im2col_nhwc() local
DCol2Im.cpp43 …const size_t batches = dst_shape.total_size() / (dst_shape.x() * dst_shape.y() * dst_shape.z()); in col2im() local
DChannelShuffle.cpp47 const int batches = src.shape()[3]; in channel_shuffle() local
DReverse.cpp49 const unsigned int batches = src.shape()[3]; in reverse() local
/external/ComputeLibrary/examples/
Dgraph_lenet.cpp66 unsigned int batches = 4; /** Number of batches */ in do_setup() local
/external/armnn/src/backends/reference/workloads/
DDepthToSpace.cpp28 const unsigned int batches = inputShape[0]; in DepthToSpace() local
/external/tensorflow/tensorflow/lite/kernels/
Dsvdf_test.cc131 BaseSVDFOpModel(int batches, int units, int input_size, int memory_size, in BaseSVDFOpModel()
206 HybridSVDFOpModel(int batches, int units, int input_size, int memory_size, in HybridSVDFOpModel()
470 IntegerSVDFOpModel(int batches, int units, int input_size, int memory_size, in IntegerSVDFOpModel()
/external/ComputeLibrary/utils/
DCommonGraphOptions.h155 SimpleOption<int> *batches; /**< Number of batches */ variable
/external/armnn/src/armnn/layers/
DFullyConnectedLayer.cpp42 unsigned int batches = inputShape[0]; in InferOutputShapes() local

12345