| /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_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_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()
186 EigenApprox(input_backward(id, i, j, k), expected); in TEST()
249 EigenApprox(input_backward(k, j, i, id), expected); in TEST()
298 EigenApprox(input_backward(id, i, j), expected); in TEST()
349 EigenApprox(input_backward(j, i, id), expected); in TEST()
419 EigenApprox(input_backward(id, i, j, k), expected); in TEST()
490 EigenApprox(input_backward(k, j, i, id), expected); in TEST()
545 EigenApprox(input_backward(id, 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_softmax_test.cc | 22 void EigenApprox(float a, float b) { in EigenApprox() function
59 EigenApprox(result(i, j), reference(i, j)); in TEST()
|