Searched refs:add_testcase (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 408 auto add_testcase = [&](std::vector<int64> bounds, int64 feature_index, in BuildBatchNormTestParams() local
421 add_testcase({2, 2, 2, 2}, 0, 100.2f, 200.0f); in BuildBatchNormTestParams()
422 add_testcase({2, 2, 2, 2}, 3, 300.f, 400.0f); in BuildBatchNormTestParams()
424 add_testcase({1, 10, 1, 1}, 0, 10.1f, 20.1f); in BuildBatchNormTestParams()
425 add_testcase({10, 10, 10, 10}, 1, 3.14f, 314.15f); in BuildBatchNormTestParams()
426 add_testcase({10, 10, 10, 10}, 2, 666.6f, 777.7f); in BuildBatchNormTestParams()
427 add_testcase({10, 10, 10, 10}, 1, -666.6f, 777.7f); in BuildBatchNormTestParams()
428 add_testcase({10, 10, 10, 10}, 2, 0.f, 777.7f); in BuildBatchNormTestParams()
429 add_testcase({1, 1, 10, 130}, 2, 0.f, 777.7f); in BuildBatchNormTestParams()
430 add_testcase({1, 1, 130, 11}, 2, 0.f, 777.7f); in BuildBatchNormTestParams()
[all …]
|