| 4743 | def build_graph(parameters): |
| 4744 | """Build the batch_matmul op testing graph.""" |
| 4745 | def _build_graph(): |
| 4746 | input_tensor1 = tf.placeholder( |
| 4747 | dtype=parameters["dtype"], shape=parameters["shape"][0]) |
| 4748 | input_tensor2 = tf.placeholder( |
| 4749 | dtype=parameters["dtype"], shape=parameters["shape"][1]) |
| 4750 | # Should be unrolled and replaced with fully_connected ops in the end. |
| 4751 | out = tf.matmul( |
| 4752 | input_tensor1, |
| 4753 | input_tensor2, |
| 4754 | transpose_a=parameters["shape"][2], |
| 4755 | transpose_b=parameters["shape"][3]) |
| 4756 | return [input_tensor1, input_tensor2], [out] |
| 4757 | if parameters["forward_compatibility_test"]: |
| 4758 | # This is hardcoded to the date after MatMulV2 is activated. |
| 4759 | # TODO(b/130887442): Improve the forward compatibility tests for every |