(weights_shape, values_shape)
| 31 | |
| 32 | |
| 33 | def _has_valid_dims(weights_shape, values_shape): |
| 34 | with ops.name_scope( |
| 35 | None, "has_invalid_dims", (weights_shape, values_shape)) as scope: |
| 36 | values_shape_2d = array_ops.expand_dims(values_shape, -1) |
| 37 | valid_dims = array_ops.concat( |
| 38 | (values_shape_2d, array_ops.ones_like(values_shape_2d)), axis=1) |
| 39 | weights_shape_2d = array_ops.expand_dims(weights_shape, -1) |
| 40 | invalid_dims = sets.set_difference(weights_shape_2d, valid_dims) |
| 41 | num_invalid_dims = array_ops.size( |
| 42 | invalid_dims.values, name="num_invalid_dims") |
| 43 | return math_ops.equal(0, num_invalid_dims, name=scope) |
| 44 | |
| 45 | |
| 46 | def _has_valid_nonscalar_shape( |
no test coverage detected