(input, dim, keepdims=False)
| 160 | |
| 161 | |
| 162 | def sum(input, dim, keepdims=False): |
| 163 | if input.dtype == tf.bool: |
| 164 | input = tf.cast(input, tf.int32) |
| 165 | return tf.reduce_sum(input, axis=dim, keepdims=keepdims) |
| 166 | |
| 167 | |
| 168 | def floor_div(in1, in2): |
no outgoing calls
no test coverage detected