Lines Matching refs:num_elements
68 return ops.OpStats("flops", in_shape.num_elements() * ops_per_element)
121 return ops.OpStats("flops", in_shape.num_elements() * 3 - 1)
144 return ops.OpStats("flops", out_shape.num_elements() * ops_per_element)
247 num_flops = (in_shape.num_elements() * reduce_flops
248 + out_shape.num_elements() * (finalize_flops - reduce_flops))
321 return ops.OpStats("flops", kernel_area * out_shape.num_elements())
350 kernel_area * out_backprop_shape.num_elements() * 2)
377 max_pool_ops = kernel_area * orig_out_shape.num_elements()
378 return ops.OpStats("flops", max_pool_ops + orig_out_shape.num_elements())
408 (2 * out_shape.num_elements()
409 * kernel_shape.num_elements()
431 (2 * image_shape.num_elements()
432 * kernel_shape.num_elements()
447 return ops.OpStats("flops", in_shape.num_elements() * (len(node.input) - 1))
