Searched refs:numBatches (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | quant_basic_lstm_test.cc | 32 QuantizedLSTMOpModel(int numBatches, int inputSize, float weightsScale, in QuantizedLSTMOpModel() argument
45 std::vector<int> input_shape{numBatches, inputSize}; in QuantizedLSTMOpModel()
46 std::vector<int> output_shape{numBatches, outputSize}; in QuantizedLSTMOpModel()
48 std::vector<int> state_shape{numBatches, outputSize}; in QuantizedLSTMOpModel()
169 const int numBatches = input.size(); in VerifyGoldens() local
170 ASSERT_GT(numBatches, 0); in VerifyGoldens()
177 for (int b = 0; b < numBatches; ++b) { in VerifyGoldens()
187 for (int b = 0; b < numBatches; ++b) { in VerifyGoldens()
201 const int numBatches = 2; in TEST_P() local
211 numBatches, inputSize, weightsScale, weightsZeroPoint, outputSize, in TEST_P()
|
| /external/swiftshader/src/Device/ |
| D | Renderer.cpp | 263 draw->numBatches = (count + draw->numPrimitivesPerBatch - 1) / draw->numPrimitivesPerBatch; in draw()
487 auto const numBatches = draw->numBatches; in run() local
496 for(unsigned int batchId = 0; batchId < numBatches; batchId++) in run()
|
| D | Renderer.hpp | 154 unsigned int numBatches; member
|