(pfor_input)
| 2181 | |
| 2182 | @RegisterPFor("BatchMatMulV2") |
| 2183 | def _convert_batch_mat_mul_v2(pfor_input): |
| 2184 | pfor_input.expanddim_inputs_for_broadcast() |
| 2185 | x = pfor_input.input(0)[0] |
| 2186 | y = pfor_input.input(1)[0] |
| 2187 | adj_x = pfor_input.get_attr("adj_x") |
| 2188 | adj_y = pfor_input.get_attr("adj_y") |
| 2189 | |
| 2190 | output = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y) |
| 2191 | return wrap(output, True) |
| 2192 | |
| 2193 | |
| 2194 | @RegisterPForWithArgs("Sum", math_ops.reduce_sum) |
nothing calls this directly
no test coverage detected