| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_activations_test.cc | 22 void EigenApprox(float a, float b) { in EigenApprox() function
44 EigenApprox(result(d, r, c, b), (1 - val) * val); in TEST()
68 EigenApprox(result(d, r, c, b), 1 - (val * val)); in TEST()
92 EigenApprox(result(d, r, c, b), in TEST()
|
| D | eigen_backward_spatial_convolutions_test.cc | 23 void EigenApprox(float a, float b) { in EigenApprox() function
72 EigenApprox(input_backward(id, i, j), expected); in TEST()
124 EigenApprox(input_backward(j, i, id), expected); in TEST()
172 EigenApprox(input_backward(id, i, j), expected); in TEST()
223 EigenApprox(input_backward(j, i, id), expected); in TEST()
277 EigenApprox(input_backward(id, i, j, b), expected); in TEST()
333 EigenApprox(input_backward(b, j, i, id), expected); in TEST()
389 EigenApprox(input_backward(id, i, j, b), expected); in test_batched_strided_spatial_convolution_backward_input_valid()
466 EigenApprox(input_backward(b, j, i, id), expected); in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
539 EigenApprox(kernel(od, id, r, c), expected); in TEST()
[all …]
|
| D | eigen_spatial_convolutions_test.cc | 25 #define EigenApprox(a, b) \ macro
68 EigenApprox(result(od, i, j), expected); in TEST()
113 EigenApprox(result(j, i, od), expected); in TEST()
149 EigenApprox(result(od, i, j, b), expected); in TEST()
186 EigenApprox(result(b, j, i, od), expected); in TEST()
238 EigenApprox(result(od, i, j, b), expected); in TEST()
296 EigenApprox(result(od, i, j, b), expected); in TEST()
352 EigenApprox(result(b, j, i, od), expected); in TEST()
400 EigenApprox(result(od, i, j, b), expected); in TEST()
448 EigenApprox(result(od, i, j, b), expected); in TEST()
[all …]
|
| D | eigen_pooling_test.cc | 22 void EigenApprox(float a, float b) { in EigenApprox() function
68 EigenApprox(result(d, i, j, b), expected); in TEST()
117 EigenApprox(result(b, j, i, d), expected); in TEST()
176 EigenApprox(result(d, i, j, k, b), expected); in TEST()
236 EigenApprox(result(b, k, j, i, d), expected); in TEST()
298 EigenApprox(result(d, i, j, k, b), expected); in TEST()
360 EigenApprox(result(b, k, j, i, d), expected); in TEST()
438 EigenApprox(result(d, i, j, k, b), expected); in TEST()
516 EigenApprox(result(b, k, j, i, d), expected); in TEST()
565 EigenApprox(result(d, i, j, b), expected); in TEST()
[all …]
|
| D | eigen_attention_test.cc | 22 void EigenApprox(float a, float b) { in EigenApprox() function
56 EigenApprox(result(d, r, c, b), input(d, source_r, source_c, b)); in TEST()
99 EigenApprox(result(d, r, c, b), input(d, source_r, source_c, b)); in TEST()
|
| D | eigen_backward_cuboid_convolutions_test.cc | 23 void EigenApprox(float a, float b) { in EigenApprox() function
85 EigenApprox(input_backward(id, i, j, k), expected); in TEST()
148 EigenApprox(input_backward(k, j, i, id), expected); in TEST()
219 EigenApprox(input_backward(id, i, j, k), expected); in TEST()
290 EigenApprox(input_backward(k, j, i, id), expected); in TEST()
356 EigenApprox(input_backward(id, i, j, k, b), expected); in TEST()
423 EigenApprox(input_backward(b, k, j, i, id), expected); in TEST()
489 EigenApprox(kernel(od, id, p, r, c), expected); in TEST()
555 EigenApprox(kernel(c, r, p, id, od), expected); in TEST()
622 EigenApprox(kernel_backward(od, id, p, r, c), expected); in TEST()
[all …]
|