| /external/XNNPACK/test/ |
| D | f32-hswish.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
71 .batch_size(8) in TEST()
[all …]
|
| D | f32-sigmoid.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
71 .batch_size(8) in TEST()
[all …]
|
| D | f32-vadd.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vmin.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vsub.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vdiv.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vmax.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vmul.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vsubc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vminc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vmaxc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vdivc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vmulc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vaddc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vrsubc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | f32-vrdivc.cc | 24 .batch_size(4) in TEST()
30 for (size_t batch_size = 8; batch_size < 40; batch_size += 4) { in TEST() local
32 .batch_size(batch_size) in TEST()
39 for (size_t batch_size = 1; batch_size < 4; batch_size++) { in TEST() local
41 .batch_size(batch_size) in TEST()
48 for (size_t batch_size = 5; batch_size < 8; batch_size++) { in TEST() local
50 .batch_size(batch_size) in TEST()
57 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
59 .batch_size(batch_size) in TEST()
67 for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) { in TEST() local
[all …]
|
| D | fully-connected-nc.cc | 16 .batch_size(1) in TEST()
25 .batch_size(1) in TEST()
35 .batch_size(1) in TEST()
45 .batch_size(1) in TEST()
55 .batch_size(1) in TEST()
66 .batch_size(1) in TEST()
76 .batch_size(1) in TEST()
85 .batch_size(12) in TEST()
94 .batch_size(12) in TEST()
104 .batch_size(12) in TEST()
[all …]
|
| D | softmax-nc.cc | 16 .batch_size(1) in TEST()
24 .batch_size(1) in TEST()
33 .batch_size(1) in TEST()
43 .batch_size(1) in TEST()
49 .batch_size(1) in TEST()
58 .batch_size(1) in TEST()
64 .batch_size(1) in TEST()
70 .batch_size(1) in TEST()
80 .batch_size(1) in TEST()
93 .batch_size(1) in TEST()
[all …]
|
| D | channel-shuffle-nc.cc | 17 .batch_size(1) in TEST()
28 .batch_size(1) in TEST()
39 .batch_size(1) in TEST()
51 .batch_size(1) in TEST()
63 .batch_size(3) in TEST()
74 .batch_size(3) in TEST()
85 .batch_size(3) in TEST()
97 .batch_size(3) in TEST()
109 .batch_size(3) in TEST()
121 .batch_size(3) in TEST()
[all …]
|
| D | add-nc.cc | 17 .batch_size(1) in TEST()
27 .batch_size(1) in TEST()
38 .batch_size(1) in TEST()
50 .batch_size(1) in TEST()
63 .batch_size(1) in TEST()
76 .batch_size(1) in TEST()
89 .batch_size(1) in TEST()
102 .batch_size(1) in TEST()
115 .batch_size(1) in TEST()
127 .batch_size(3) in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 26 int input_size, int num_units, int batch_size, in RnnBatchStep() argument
34 batch_size, output_batch_leading_dim, activation, in RnnBatchStep()
43 int batch_size, int output_batch_leading_dim, in RnnBatchStep() argument
50 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
55 input_weights_ptr, num_units, input_size, input_ptr_batch, batch_size, in RnnBatchStep()
62 batch_size, output_ptr_batch, /*result_stride=*/1); in RnnBatchStep()
68 batch_size, output_ptr_batch, /*result_stride=*/1); in RnnBatchStep()
72 output_ptr_batch, num_units * batch_size, activation, output_ptr_batch); in RnnBatchStep()
73 std::copy_n(output_ptr_batch, num_units * batch_size, in RnnBatchStep()
77 for (int k = 0; k < batch_size; k++) { in RnnBatchStep()
[all …]
|
| /external/igt-gpu-tools/tests/i915/ |
| D | gem_exec_big.c | 128 static void execN(int fd, uint32_t handle, uint64_t batch_size, unsigned flags, char *ptr) in execN() argument
134 uint64_t n, nreloc = batch_size >> 12; in execN()
204 uint64_t batch_size, max, ggtt_max, reloc_ofs; in exhaustive() local
210 for (batch_size = 4096; batch_size <= max; ) { in exhaustive()
214 handle = gem_create(fd, batch_size); in exhaustive()
218 ptr = __gem_mmap__cpu(fd, handle, 0, batch_size, PROT_READ); in exhaustive()
220 ptr = __gem_mmap__wc(fd, handle, 0, batch_size, PROT_READ); in exhaustive()
224 igt_debug("Forwards (%lld)\n", (long long)batch_size); in exhaustive()
225 for (reloc_ofs = 4096; reloc_ofs < batch_size; reloc_ofs += 4096) { in exhaustive()
227 (long long)batch_size, (long long)reloc_ofs); in exhaustive()
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | make_batched_features_dataset_test.py | 44 for batch_size in [1, 2]:
52 batch_size=batch_size))
54 batch_size, 0, num_epochs=num_epochs, label_key_provided=True)
64 batch_size=batch_size))
66 batch_size, 1, num_epochs=num_epochs, label_key_provided=True)
76 batch_size=batch_size))
78 batch_size, num_epochs=num_epochs, label_key_provided=True)
86 batch_size=batch_size))
87 self.verify_records(batch_size, num_epochs=num_epochs)
114 for batch_size in [1, 2]:
[all …]
|
| D | make_tf_record_dataset_test.py | 37 def _read_test(self, batch_size, num_epochs, file_index=None, argument
53 batch_size=batch_size,
60 batch_size,
71 for batch_size in [1, 2]:
74 self._read_test(batch_size, num_epochs, 0)
77 self._read_test(batch_size, num_epochs, 1)
80 self._read_test(batch_size, num_epochs)
83 self._read_test(batch_size, num_epochs, num_parallel_reads=8)
87 for batch_size in [1, 2, 10]:
90 self._read_test(batch_size, num_epochs, 0, drop_final_batch=True)
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | svdf_test.cc | 124 void ValidateSVDFGoldens(const int batch_size, const int num_units, in ValidateSVDFGoldens() argument
183 golden_input_data_size / sizeof(float) / (input_size * batch_size); in ValidateSVDFGoldens()
185 float* input_batch_start = golden_input_data + i * input_size * batch_size; in ValidateSVDFGoldens()
186 float* input_batch_end = input_batch_start + input_size * batch_size; in ValidateSVDFGoldens()
192 int golden_idx = i * batch_size * num_units; in ValidateSVDFGoldens()
193 for (int j = golden_idx; j < golden_idx + batch_size * num_units; ++j) { in ValidateSVDFGoldens()
205 void ValidateIntegerSVDFGoldens(const int batch_size, const int num_units, in ValidateIntegerSVDFGoldens() argument
249 golden_input_data_size / sizeof(int8_t) / (input_size * batch_size); in ValidateIntegerSVDFGoldens()
251 int8_t* input_batch_start = golden_input_data + i * input_size * batch_size; in ValidateIntegerSVDFGoldens()
252 int8_t* input_batch_end = input_batch_start + input_size * batch_size; in ValidateIntegerSVDFGoldens()
[all …]
|