Home
last modified time | relevance | path

Searched refs:kBatchSize (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/image/
Dscale_and_translate_op_test.cc273 constexpr int64 kBatchSize = 2; in TEST_F() local
278 SetCheckerboardImageInput(kBatchSize, kNumRowSquares, kNumColSquares, in TEST_F()
289 constexpr int64 kBatchSize = 2; in TEST_F() local
294 SetCheckerboardImageInput(kBatchSize, kNumRowSquares, kNumColSquares, in TEST_F()
305 constexpr int64 kBatchSize = 2; in TEST_F() local
310 SetCheckerboardImageInput(kBatchSize, kNumRowSquares, kNumColSquares, in TEST_F()
321 constexpr int64 kBatchSize = 2; in TEST_F() local
326 SetCheckerboardImageInput(kBatchSize, kNumRowSquares, kNumColSquares, in TEST_F()
338 constexpr int64 kBatchSize = 2; in TEST_F() local
343 SetCheckerboardImageInput(kBatchSize, kNumRowSquares, kNumColSquares, in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization_test.cc52 static constexpr int kBatchSize = 50; member in xla::cpu::ConvCanonicalizationTest
64 kInputFeatureCount, kBatchSize, kInputSize, kInputSize)))); in TEST_F()
86 F32, {kOutputFeatureCount, kBatchSize, output_size, output_size}), in TEST_F()
127 kBatchSize, kInputSize, kInputSize, kInputFeatureCount)))); in TEST_F()
149 F32, {kBatchSize, output_size, output_size, kOutputFeatureCount}), in TEST_F()
/external/perfetto/src/trace_processor/containers/
Dbit_vector_iterators.h165 if (PERFETTO_UNLIKELY(set_bit_index_ % kBatchSize == 0)) in Next()
168 SetIndex(batch_[set_bit_index_ % kBatchSize]); in Next()
179 static constexpr uint32_t kBatchSize = 1024;
193 std::array<uint32_t, kBatchSize> batch_;
Dbit_vector_iterators.cc87 PERFETTO_DCHECK(set_bit_index_ % kBatchSize == 0); in ReadSetBitBatch()
118 uint32_t batch_idx = set_bit_count_until_i++ % kBatchSize; in ReadSetBitBatch()
123 if (PERFETTO_UNLIKELY(batch_idx == kBatchSize - 1)) in ReadSetBitBatch()
/external/webrtc/rtc_base/
Drate_statistics_unittest.cc181 const int kBatchSize = 10; in TEST_F() local
182 for (int i = 0; i <= kWindowMs; i += kBatchSize) in TEST_F()
183 stats_.Update(kBatchSize, now_ms += kBatchSize); in TEST_F()
196 for (int i = 0; i < kWindowMs / 2; i += kBatchSize) in TEST_F()
197 stats_.Update(kBatchSize * 2, now_ms += kBatchSize); in TEST_F()
/external/libchrome/base/message_loop/
Dmessage_pump_perftest.cc51 for (size_t i = 0; i < kBatchSize; ++i) { in Schedule()
149 min_batch_time.InMicroseconds() / static_cast<double>(kBatchSize), in ScheduleWork()
156 max_batch_time.InMicroseconds() / static_cast<double>(kBatchSize), in ScheduleWork()
190 static const size_t kBatchSize = 1000; member in base::ScheduleWorkTest
/external/icing/icing/tokenization/reverse_jni/
Dreverse-jni-break-iterator.cc36 static constexpr int kBatchSize = 100; variable
116 is_almost_done_ = break_indices_cache_.size() < kBatchSize; in Next()
159 jni_cache_->breakiterator_next, kBatchSize), in FetchNextBatch()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc44 constexpr int kBatchSize = 32; variable
107 {kBatchSize, kHeight, kWidth, kDepthIn}, data_type); in SimpleConv2D()
132 {kBatchSize, kHeight, kWidth, kDepthIn}, data_type); in CreateSimpleFusedBatchNorm()
157 {kBatchSize, kHeight, kWidth, kDepthIn}, DT_FLOAT); in CreateSimpleMaxPoolGrad()
160 {kBatchSize, kOutHeight, kOutWidth, kDepthIn}, DT_FLOAT); in CreateSimpleMaxPoolGrad()
163 {kBatchSize, use_grad_grad ? kHeight : kOutHeight, in CreateSimpleMaxPoolGrad()
201 {kBatchSize, kHeight, kWidth, kDepthIn}, data_type); in CreateSimpleConv2DBackpropFilter()
204 {kBatchSize, kHeight, kWidth, kDepthOut}, data_type); in CreateSimpleConv2DBackpropFilter()
234 {kBatchSize, kHeight, kWidth, kDepthIn}); in CreateSimpleConv2DBackpropInput()
237 {kBatchSize, kHeight, kWidth, kDepthIn}, data_type); in CreateSimpleConv2DBackpropInput()
[all …]
/external/llvm-project/compiler-rt/lib/sanitizer_common/
Dsanitizer_allocator_primary32.h110 static const uptr kBatchSize = sizeof(TransferBatch); variable
111 COMPILER_CHECK((kBatchSize & (kBatchSize - 1)) == 0);
112 COMPILER_CHECK(kBatchSize == SizeClassMap::kMaxNumCachedHint * sizeof(uptr));
116 kBatchSize : SizeClassMap::Size(class_id); in ClassIdToSize()
/external/angle/third_party/abseil-cpp/absl/synchronization/
Dmutex_benchmark.cc132 static constexpr int kBatchSize = 1000; in BM_MutexEnqueue() local
133 while (state.KeepRunningBatch(kBatchSize)) { in BM_MutexEnqueue()
135 for (int i = 0; i < kBatchSize; i++) { in BM_MutexEnqueue()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_test.cc1677 constexpr int kBatchSize = 2; in TEST() local
1680 VectorBatchVectorAdd(input, kVectorSize, kBatchSize, output.data()); in TEST()
1687 constexpr int kBatchSize = 3; in TEST() local
1689 std::vector<float> output(kVectorSize * kBatchSize); in TEST()
1690 VectorBatchVectorAssign(input, kVectorSize, kBatchSize, output.data()); in TEST()
1747 constexpr int kBatchSize = 4; in TEST() local
1779 batch_vector.data(), kBatchSize, in TEST()
1802 constexpr int kBatchSize = 4; in TEST() local
1829 kBatchSize, output.data()); in TEST()
1862 constexpr int kBatchSize = 4; in TEST() local
[all …]
/external/openscreen/third_party/abseil/src/absl/container/
Dinlined_vector_benchmark.cc383 constexpr size_t kBatchSize = 100; variable
431 std::array<InlVec<T>, kBatchSize> vector_batch{}; in BatchedBenchmark()
433 while (state.KeepRunningBatch(kBatchSize)) { in BatchedBenchmark()
436 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
443 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
519 std::array<VecT, kBatchSize> vector_batch{}; in BM_ConstructFromMove()
587 std::array<VecT, kBatchSize> vector_batch{}; in BM_AssignFromMove()
788 std::array<VecT, kBatchSize> vector_batch{}; in BM_Swap()
/external/abseil-cpp/absl/container/
Dinlined_vector_benchmark.cc383 constexpr size_t kBatchSize = 100; variable
431 std::array<InlVec<T>, kBatchSize> vector_batch{}; in BatchedBenchmark()
433 while (state.KeepRunningBatch(kBatchSize)) { in BatchedBenchmark()
436 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
443 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
519 std::array<VecT, kBatchSize> vector_batch{}; in BM_ConstructFromMove()
587 std::array<VecT, kBatchSize> vector_batch{}; in BM_AssignFromMove()
788 std::array<VecT, kBatchSize> vector_batch{}; in BM_Swap()
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/container/
Dinlined_vector_benchmark.cc383 constexpr size_t kBatchSize = 100; variable
431 std::array<InlVec<T>, kBatchSize> vector_batch{}; in BatchedBenchmark()
433 while (state.KeepRunningBatch(kBatchSize)) { in BatchedBenchmark()
436 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
443 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
519 std::array<VecT, kBatchSize> vector_batch{}; in BM_ConstructFromMove()
587 std::array<VecT, kBatchSize> vector_batch{}; in BM_AssignFromMove()
788 std::array<VecT, kBatchSize> vector_batch{}; in BM_Swap()
/external/angle/third_party/abseil-cpp/absl/container/
Dinlined_vector_benchmark.cc383 constexpr size_t kBatchSize = 100; variable
431 std::array<InlVec<T>, kBatchSize> vector_batch{}; in BatchedBenchmark()
433 while (state.KeepRunningBatch(kBatchSize)) { in BatchedBenchmark()
436 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
443 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
519 std::array<VecT, kBatchSize> vector_batch{}; in BM_ConstructFromMove()
609 std::array<VecT, kBatchSize> vector_batch{}; in BM_AssignFromMove()
810 std::array<VecT, kBatchSize> vector_batch{}; in BM_Swap()
/external/webrtc/third_party/abseil-cpp/absl/container/
Dinlined_vector_benchmark.cc383 constexpr size_t kBatchSize = 100; variable
431 std::array<InlVec<T>, kBatchSize> vector_batch{}; in BatchedBenchmark()
433 while (state.KeepRunningBatch(kBatchSize)) { in BatchedBenchmark()
436 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
443 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
519 std::array<VecT, kBatchSize> vector_batch{}; in BM_ConstructFromMove()
587 std::array<VecT, kBatchSize> vector_batch{}; in BM_AssignFromMove()
788 std::array<VecT, kBatchSize> vector_batch{}; in BM_Swap()
/external/libtextclassifier/abseil-cpp/absl/container/
Dinlined_vector_benchmark.cc383 constexpr size_t kBatchSize = 100; variable
431 std::array<InlVec<T>, kBatchSize> vector_batch{}; in BatchedBenchmark()
433 while (state.KeepRunningBatch(kBatchSize)) { in BatchedBenchmark()
436 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
443 for (size_t i = 0; i < kBatchSize; ++i) { in BatchedBenchmark()
519 std::array<VecT, kBatchSize> vector_batch{}; in BM_ConstructFromMove()
587 std::array<VecT, kBatchSize> vector_batch{}; in BM_AssignFromMove()
788 std::array<VecT, kBatchSize> vector_batch{}; in BM_Swap()
/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc50 constexpr int kBatchSize = 1; variable
177 {kBatchSize, num_detected_boxes, kNumCoordBox}); in Prepare()
185 SetTensorSizes(context, detection_classes, {kBatchSize, num_detected_boxes}); in Prepare()
193 SetTensorSizes(context, detection_scores, {kBatchSize, num_detected_boxes}); in Prepare()
286 TF_LITE_ENSURE_EQ(context, input_box_encodings->dims->data[0], kBatchSize); in DecodeCenterSizeBoxes()
757 kBatchSize); in NonMaxSuppressionMultiClass()
792 TF_LITE_ENSURE(context, (kBatchSize == 1)); in Eval()
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op.h28 static constexpr const char* const kBatchSize = "batch_size"; variable
Dbatch_dataset_op.h27 static constexpr const char* const kBatchSize = "batch_size"; variable
Dpadded_batch_dataset_op.h27 static constexpr const char* const kBatchSize = "batch_size"; variable
Dbatch_dataset_op.cc39 /* static */ constexpr const char* const BatchDatasetOp::kBatchSize; member in tensorflow::data::BatchDatasetOp
267 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kBatchSize, &batch_size)); in MakeDataset()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.h33 static constexpr const char* const kBatchSize = "batch_size"; variable
/external/compiler-rt/lib/sanitizer_common/
Dsanitizer_coverage_libcdep.cc479 static const uptr kBatchSize = 8; in Update8bitCounterBitsetAndClearCounters() local
480 CHECK_EQ(reinterpret_cast<uptr>(bitset) % kBatchSize, 0); in Update8bitCounterBitsetAndClearCounters()
485 CHECK_EQ(cur % kBatchSize, 0); in Update8bitCounterBitsetAndClearCounters()
486 CHECK_EQ(reinterpret_cast<uptr>(c) % kBatchSize, 0); in Update8bitCounterBitsetAndClearCounters()
492 for (uptr j = 0; j < n; j += kBatchSize, cur += kBatchSize) { in Update8bitCounterBitsetAndClearCounters()
501 for (uptr k = 0; k < kBatchSize; k++) { in Update8bitCounterBitsetAndClearCounters()
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddetection_postprocess.cc54 constexpr int kBatchSize = 1; variable
274 TF_LITE_ENSURE_EQ(context, input_box_encodings->dims->data[0], kBatchSize); in DecodeCenterSizeBoxes()
731 kBatchSize); in NonMaxSuppressionMultiClass()
769 TF_LITE_ENSURE(context, (kBatchSize == 1)); in Eval()

12