Searched refs:get_matmul (Results 1 – 1 of 1) sorted by relevance
2474 const std::function<NodeDef(DataType, bool, bool)>& get_matmul, in TestMatMulHelper() argument2481 NodeDef node_def = get_matmul(DT_INT32, false, false); in TestMatMulHelper()2495 NodeDef node_def = get_matmul(DT_FLOAT, transpose_a, transpose_b); in TestMatMulHelper()2529 NodeDef node_def = get_matmul(DT_FLOAT, /*transpose_a=*/false, transpose_b); in TestMatMulHelper()