Home
last modified time | relevance | path

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

1234

/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
Ddepthwise_conv.h51 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvPerChannel() local
148 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvPerChannel() local
231 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvHybridPerChannel() 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/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/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/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_batch_cache.h51 struct fd_batch *batches[32]; member
/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/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
DReverse.cpp49 const unsigned int batches = src.shape()[3]; in reverse() local
DChannelShuffle.cpp47 const int batches = src.shape()[3]; in channel_shuffle() local
DCol2Im.cpp43 …const size_t batches = dst_shape.total_size() / (dst_shape.x() * dst_shape.y() * dst_shape.z()); in col2im() 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/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/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
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h139 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DilatedIm2col() local
225 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col() local
261 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col() local
388 const int batches = MatchingDim(input_shape, 0, im2col_shape, 0); in Im2col3D() local
428 const int batches = MatchingDim(input_shape, 0, im2col_shape, 0); in DilatedIm2col3D() local

1234