Home
last modified time | relevance | path

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

123456789

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A 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
H A 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
H A Dconv.h47 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv() local
137 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv() local
224 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in HybridConvPerChannel() local
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A 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
H A 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
H A 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
H A Dconv.h52 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in ConvPerChannel() local
163 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in ConvPerChannel() local
/aosp_15_r20/external/igt-gpu-tools/tests/i915/
H A 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
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp333 uint32_t batches = getSizeOfDimension(input, 0); in depthToSpacePrepare() local
352 uint32_t batches = getSizeOfDimension(input, 0); in spaceToDepthPrepare() local
447 uint32_t batches = getSizeOfDimension(input, 0); in batchToSpacePrepare() local
480 uint32_t batches = getSizeOfDimension(input, 0); in spaceToBatchPrepare() local
646 uint32_t batches = getSizeOfDimension(input, 0); in groupedConvPrepare() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A 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
H A Dresize_bilinear.h265 inline void ResizeBilinear888Uint8(int32 batches, int32 input_height, in ResizeBilinear888Uint8()
1492 inline void ResizeBilinear2x2(int32 batches, int32 input_height, in ResizeBilinear2x2()
1512 int32 batches, int32 input_height, int32 input_width, int32 depth, in ResizeBilinearGeneric()
1565 int32 batches, int32 input_height, int32 input_width, int32 depth, in ResizeBilinearGenericSmallChannel()
1626 int32 batches = MatchingDim(input_shape, 0, output_shape, 0); in ResizeBilinear() local
1677 int32 batches = MatchingDim(input_shape, 0, output_shape, 0); in ResizeBilinear() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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()
H A Dfully_connected_test.cc138 TfLiteRegistration* registration, int units, int batches, in BaseFullyConnectedOpModel()
283 TfLiteRegistration* registration, int units, int batches, in QuantizedFullyConnectedOpModel()
358 TfLiteRegistration* registration, int units, int batches, in PerChannelQuantizedFullyConnectedOpModel()
404 HybridFullyConnectedOpModel(int units, int batches, const TensorData& input, in HybridFullyConnectedOpModel()
1054 int batches, FullyConnectedOptionsWeightsFormat weights_format) { in SimpleTestQuantizedInt16OutputCase()
1542 int batches, const TensorData& input, in SparseFullyConnectedOpModel()
H A Dpooling3d.cc80 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool3D() local
160 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool3D() local
305 int batches = input->dims->data[0]; in GenericPrepare() local
H A Dbasic_rnn_test.cc173 RNNOpModel(int batches, int units, int size, in RNNOpModel()
235 HybridRNNOpModel(int batches, int units, int size, TensorType tensor_type, in HybridRNNOpModel()
/aosp_15_r20/external/pytorch/aten/src/ATen/benchmarks/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/perception/
H A 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
H A Dmax_pool_with_argmax.cc42 const int32_t batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
179 int batches = input->dims->data[0]; in Prepare() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DTransposeConvolution2dTestImpl.cpp229 constexpr unsigned int batches = 1u; in SimpleTransposeConvolution2dTest() local
318 constexpr unsigned int batches = 1u; in PaddedTransposeConvolution2dTest() local
409 constexpr unsigned int batches = 1u; in StridedTransposeConvolution2dTest() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A 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
H A Dpooling.h53 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
163 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_quantized_dot_ops.cc61 const int64_t batches = output.dim_size(0); in DotWithAccFunctionAndOutputFunction() local
137 const int64_t batches = lhs.dim_size(0); in EvalHybridDot() local

123456789