Searched refs:get_batch_matmul_nodedef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2639 auto get_batch_matmul_nodedef = [](DataType dtype, bool transpose_a, in TEST_F() local
2654 NodeDef node_def = get_batch_matmul_nodedef(DT_FLOAT, false, false); in TEST_F()
2665 NodeDef node_def = get_batch_matmul_nodedef(DT_FLOAT, true, false); in TEST_F()
2677 get_batch_matmul_nodedef(DT_FLOAT, transpose_a, transpose_b); in TEST_F()
2705 TestMatMulHelper(this, get_batch_matmul_nodedef, "BatchMatMul"); in TEST_F()
|