Searched refs:add_matrix_matrix_dot_test (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 396 auto add_matrix_matrix_dot_test = [&](int m, int k, int n) { in CreateDotTestParameters() local
407 add_matrix_matrix_dot_test(/*m=*/1, /*k=*/23, /*n=*/42); in CreateDotTestParameters()
408 add_matrix_matrix_dot_test(/*m=*/23, /*k=*/1, /*n=*/42); in CreateDotTestParameters()
409 add_matrix_matrix_dot_test(/*m=*/23, /*k=*/42, /*n=*/1); in CreateDotTestParameters()
410 add_matrix_matrix_dot_test(/*m=*/1, /*k=*/23, /*n=*/1); in CreateDotTestParameters()
411 add_matrix_matrix_dot_test(/*m=*/1, /*k=*/1, /*n=*/1); in CreateDotTestParameters()
412 add_matrix_matrix_dot_test(/*m=*/12, /*k=*/117, /*n=*/7); in CreateDotTestParameters()
413 add_matrix_matrix_dot_test(/*m=*/270, /*k=*/270, /*n=*/520); in CreateDotTestParameters()
414 add_matrix_matrix_dot_test(/*m=*/260, /*k=*/3, /*n=*/520); in CreateDotTestParameters()
|