Searched refs:get_matmul_nodedef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2568 auto get_matmul_nodedef = [](DataType dtype, bool transpose_a, in TEST_F() local
2584 NodeDef node_def = get_matmul_nodedef(DT_FLOAT, true, false); in TEST_F()
2595 NodeDef node_def = get_matmul_nodedef(DT_FLOAT, false, false); in TEST_F()
2606 NodeDef node_def = get_matmul_nodedef(DT_FLOAT, true, false); in TEST_F()
2616 NodeDef node_def = get_matmul_nodedef(DT_FLOAT, false, false); in TEST_F()
2627 NodeDef node_def = get_matmul_nodedef(DT_FLOAT, false, false); in TEST_F()
2634 TestMatMulHelper(this, get_matmul_nodedef, "MatMul"); in TEST_F()
|