Home
last modified time | relevance | path

Searched defs:nBatch (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstmUtils.cpp19 uint32_t nBatch, in VectorBatchVectorAdd()
91 uint32_t nBatch, in MatrixBatchVectorMultiplyAccumulate()
115 uint32_t nBatch, in VectorBatchVectorAssign()
134 uint32_t nBatch, in VectorBatchVectorCwiseProductAccumulate()
155 uint32_t nBatch, in VectorBatchVectorCwiseProduct()
H A DLstm.cpp63 const uint32_t nBatch = inputShape[0]; in LstmImpl() local
H A DRefLstmWorkload.cpp74 const uint32_t nBatch = inputShape[0]; in Execute() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DLstmTestImpl.cpp32 uint32_t nBatch, in LstmUtilsVectorBatchVectorAddTestImpl()
88 uint32_t nBatch, in LstmUtilsMeanStddevNormalizationTestImpl()
116 uint32_t nBatch, in LstmUtilsVectorBatchVectorCwiseProductTestImpl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp1471 int64_t nBatch = qx.size(0); in _qmaxpool_2d_nhwc_kernel() local
1604 int64_t nBatch = qx.size(0); in qmaxpool_3d_nthwc_kernel() local
1816 int64_t nBatch,
1926 int64_t nBatch,
1960 int64_t nBatch,
1998 int64_t nBatch,