Searched refs:add_matrix_vector_dot_test (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | dot_operation_test.cc | 451 auto add_matrix_vector_dot_test = [&](int k, int n) { in CreateNoLayoutAssignmentDotTestParameters() local 473 add_matrix_vector_dot_test(/*k=*/8, /*n=*/8); in CreateNoLayoutAssignmentDotTestParameters() 474 add_matrix_vector_dot_test(/*k=*/130, /*n=*/8); in CreateNoLayoutAssignmentDotTestParameters() 475 add_matrix_vector_dot_test(/*k=*/8, /*n=*/130); in CreateNoLayoutAssignmentDotTestParameters() 476 add_matrix_vector_dot_test(/*k=*/290, /*n=*/130); in CreateNoLayoutAssignmentDotTestParameters() 477 add_matrix_vector_dot_test(/*k=*/1, /*n=*/1); in CreateNoLayoutAssignmentDotTestParameters() 478 add_matrix_vector_dot_test(/*k=*/1, /*n=*/16); in CreateNoLayoutAssignmentDotTestParameters() 479 add_matrix_vector_dot_test(/*k=*/1, /*n=*/4); in CreateNoLayoutAssignmentDotTestParameters() 480 add_matrix_vector_dot_test(/*k=*/1, /*n=*/3); in CreateNoLayoutAssignmentDotTestParameters() 481 add_matrix_vector_dot_test(/*k=*/3, /*n=*/16); in CreateNoLayoutAssignmentDotTestParameters() [all …]
|