Searched defs:BatchSize (Results 1 – 8 of 8) sorted by relevance
34 inline DepthToSpaceTester& BatchSize(int32_t batch_size) { in BatchSize() function40 inline int32_t BatchSize() const { return batch_size_; } in BatchSize() function
35 inline ResizeBilinearTester& BatchSize(int32_t batch_size) { in BatchSize() function41 inline int32_t BatchSize() const { return batch_size_; } in BatchSize() function
35 inline Pool2DTester& BatchSize(int32_t batch_size) { in BatchSize() function41 inline int32_t BatchSize() const { return batch_size_; } in BatchSize() function
35 inline Conv2DTester& BatchSize(int32_t batch_size) { in BatchSize() function41 inline int32_t BatchSize() const { return batch_size_; } in BatchSize() function
35 inline DepthwiseConv2DTester& BatchSize(int32_t batch_size) { in BatchSize() function41 inline int32_t BatchSize() const { return batch_size_; } in BatchSize() function
133 const ClusterBatchSize& BatchSize() const { return batch_size_; } in BatchSize() function
235 pub enum BatchSize { enum279 impl BatchSize { impl
163 enum class BatchSize { enum