Home
last modified time | relevance | path

Searched defs:numBatches (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dquant_basic_lstm_test.cc32 QuantizedLSTMOpModel(int numBatches, int inputSize, float weightsScale, in QuantizedLSTMOpModel()
169 const int numBatches = input.size(); in VerifyGoldens() local
201 const int numBatches = 2; in TEST_P() local
/external/armnn/src/backends/aclCommon/
DArmComputeTensorUtils.hpp183 uint32_t numBatches = static_cast<uint32_t>(shape[3]); in GetLinearBufferOffset() local
199 uint32_t numBatches = static_cast<uint32_t>(shape[3]); in CopyArmComputeITensorData() local
235 uint32_t numBatches = static_cast<uint32_t>(shape[3]); in CopyArmComputeITensorData() local
/external/armnn/src/armnn/layers/
DQuantizedLstmLayer.cpp86 unsigned int numBatches = inputShapes[0][0]; in InferOutputShapes() local
/external/armnn/src/backends/reference/workloads/
DTransposeConvolution2d.cpp33 const unsigned int numBatches = inputShape[0]; in TransposeConvolution2dImpl() local
DRefQLstmWorkload.cpp79 const uint32_t numBatches = inputShape[0]; in Execute() local
/external/armnn/src/backends/backendsCommon/test/
DQLstmEndToEndTestImpl.cpp38 const unsigned int numBatches = 2; in QLstmEndToEnd() local
/external/armnn/src/armnnSerializer/test/
DLstmSerializationTests.cpp1645 const unsigned int numBatches = 2; variable
1805 const unsigned int numBatches = 2; variable
1996 const unsigned int numBatches = 2; variable
/external/armnn/src/backends/backendsCommon/test/layerTests/
DLstmTestImpl.cpp1501 auto numBatches = armnn::numeric_cast<unsigned int>(inputShape[0]); in QuantizedLstmTestImpl() local
1689 unsigned int numBatches = 2; in QLstmTestImpl() local
1927 unsigned int numBatches = 2; in QLstmTestImpl1() local
2192 unsigned int numBatches = 2; in QLstmTestImpl2() local
/external/swiftshader/src/Device/
DRenderer.hpp149 unsigned int numBatches; member
DRenderer.cpp559 const auto numBatches = draw->numBatches; in run() local
/external/armnn/src/armnnTestUtils/
DCreateWorkload.hpp833 unsigned int numBatches = 2; in CreateQuantizedLstmWorkloadTest() local
990 unsigned int numBatches = 2; in CreateQLstmWorkloadTest() local
/external/armnn/src/backends/backendsCommon/
DWorkloadData.cpp3111 const uint32_t numBatches = inputInfo.GetShape()[0]; in Validate() local
3394 const uint32_t numBatches = inputInfo.GetShape()[0]; in Validate() local