(self)
| 653 | |
| 654 | @test_util.run_gpu_only |
| 655 | def testMatMulGPU(self): |
| 656 | three = constant_op.constant([[3.]]).gpu() |
| 657 | five = constant_op.constant([[5.]]).gpu() |
| 658 | product = execute( |
| 659 | b'MatMul', |
| 660 | num_outputs=1, |
| 661 | inputs=[three, five], |
| 662 | attrs=('transpose_a', False, 'transpose_b', False, 'T', |
| 663 | three.dtype.as_datatype_enum))[0] |
| 664 | self.assertAllEqual([[15.0]], product) |
| 665 | |
| 666 | def testExecuteStringAttr(self): |
| 667 | checked_three = execute( |
nothing calls this directly
no test coverage detected