Home
last modified time | relevance | path

Searched defs:batchSize (Results 1 – 25 of 115) sorted by relevance

12345

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
Dchannel-shuffle-operator-tester.h77 inline ChannelShuffleOperatorTester& batchSize(size_t batchSize) { in batchSize() argument
82 inline size_t batchSize() const { in batchSize() function
Dclamp-operator-tester.h63 inline ClampOperatorTester& batchSize(size_t batchSize) { in batchSize() argument
68 inline size_t batchSize() const { in batchSize() function
Dsoftargmax-operator-tester.h64 inline SoftArgMaxOperatorTester& batchSize(size_t batchSize) { in batchSize() function
69 inline size_t batchSize() const { in batchSize() function
Dhardsigmoid-operator-tester.h64 inline HardsigmoidOperatorTester& batchSize(size_t batchSize) { in batchSize() argument
69 inline size_t batchSize() const { in batchSize() function
Dsigmoid-operator-tester.h64 inline SigmoidOperatorTester& batchSize(size_t batchSize) { in batchSize() argument
69 inline size_t batchSize() const { in batchSize() function
Dtanh-operator-tester.h64 inline TanHOperatorTester& batchSize(size_t batchSize) { in batchSize() function
69 inline size_t batchSize() const { in batchSize() function
Dhardswish-operator-tester.h64 inline HardswishOperatorTester& batchSize(size_t batchSize) { in batchSize() function
69 inline size_t batchSize() const { in batchSize() function
Dglobal-average-pooling-operator-tester.h73 inline GlobalAveragePoolingOperatorTester& batchSize(size_t batchSize) { in batchSize() argument
78 inline size_t batchSize() const { in batchSize() function
Dleaky-relu-operator-tester.h64 inline LeakyReLUOperatorTester& batchSize(size_t batchSize) { in batchSize() function
69 inline size_t batchSize() const { in batchSize() function
Dadd-operator-tester.h78 inline AddOperatorTester& batchSize(size_t batchSize) { in batchSize() function
83 inline size_t batchSize() const { in batchSize() function
Dfully-connected-operator-tester.h46 inline FullyConnectedOperatorTester& batchSize(size_t batchSize) { in batchSize() argument
51 inline size_t batchSize() const { in batchSize() function
/external/armnn/tests/ImageTensorGenerator/
DImageTensorGenerator.hpp95 unsigned int batchSize, in PrepareImageTensor()
132 unsigned int batchSize, in PrepareImageTensor()
152 unsigned int batchSize, in PrepareImageTensor()
172 unsigned int batchSize, in PrepareImageTensor()
/external/aws-sdk-java-v2/http-clients/netty-nio-client/src/test/java/software/amazon/awssdk/http/nio/netty/internal/nrs/util/
DBatchedProducer.java41 protected final int batchSize; field in BatchedProducer
45 public BatchedProducer(long eofOn, int batchSize, long sequence, Executor executor) { in BatchedProducer()
/external/aws-sdk-java-v2/http-clients/netty-nio-client/src/test/java/software/amazon/awssdk/http/nio/netty/internal/nrs/
DHandlerPublisherVerificationTest.java49 private final int batchSize; field in HandlerPublisherVerificationTest
59 public HandlerPublisherVerificationTest(int batchSize, int publishInitial, boolean scheduled) { in HandlerPublisherVerificationTest()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
Dadd.cc20 const size_t batchSize = static_cast<size_t>(state.range(0)); in add_nc_q8() local
90 const size_t batchSize = static_cast<size_t>(state.range(0)); in add_nc_q8_inplace() local
/external/kotlinx.coroutines/kotlinx-coroutines-core/concurrent/test/flow/
DStateFlowCommonStressTest.kt18 val batchSize = Random.nextInt(1..1000) in <lambda>() constant
/external/armnn/src/backends/backendsCommon/test/layerTests/
DDivisionTestImpl.cpp28 const unsigned int batchSize = 2u; in DivisionByZeroTest() local
70 const unsigned int batchSize = 2u; in DivisionTest() local
176 const unsigned int batchSize = 2u; in DivisionFloat16Test() local
284 const unsigned int batchSize = 2u; in DivisionUint8Test() local
491 const unsigned int batchSize = 2u; in DivisionInt32Test() local
DMultiplicationTestImpl.cpp27 const unsigned int batchSize = 2u; in MultiplicationTest() local
68 const unsigned int batchSize = 2u;; in Multiplication5dTest() local
200 constexpr unsigned int batchSize = 1u; in MultiplicationUint8Test() local
536 const unsigned int batchSize = 5; in CompareMultiplicationTest() local
/external/executorch/examples/mediatek/executor_runner/llama_runner/
DLlamaRuntime.cpp48 auto addModelPath = [&](const auto& modelPaths, const size_t batchSize) { in Initialize()
97 void LlamaRuntime::SwapModel(const size_t batchSize) { in SwapModel()
/external/pytorch/aten/src/ATen/functorch/
DVmapInterpreter.h15 c10::SymInt batchSize() const { in batchSize() function
/external/pytorch/aten/src/ATen/native/
DConvolutionTBC.cpp29 auto batchSize = input_size[1]; in conv_tbc() local
76 auto batchSize = input_size[1]; in conv_tbc_backward() local
/external/pytorch/aten/src/ATen/benchmarks/
Dtensor_add.cpp6 const size_t batchSize = static_cast<size_t>(state.range(0)); in tensor_add() local
/external/perfetto/ui/src/base/
Dresizable_array_buffer_unittest.ts42 for (const batchSize of BATCH_SIZES) { constant
/external/android-nn-driver/1.0/
DFullyConnected.hpp23 unsigned int batchSize = totalInputElements / inputSize; in FlattenFullyConnectedInput() local
/external/kotlinx.coroutines/kotlinx-coroutines-core/common/test/flow/sharing/
DShareInBufferTest.kt22 val batchSize = capacity + 1 in <lambda>() constant

12345