Searched refs:sum_axis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | concatenation.cc | 64 int sum_axis = t0->dims->data[axis]; in Prepare() local
72 sum_axis += t->dims->data[axis]; in Prepare()
81 output_size->data[d] = (d == axis) ? sum_axis : t0->dims->data[d]; in Prepare()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops.py | 765 sum_axis = None if axis is None else axis[0]
766 result = math_ops.reduce_sum(result, sum_axis, keepdims=True)
|
| D | image_ops_impl.py | 3265 sum_axis = None
3277 sum_axis = [1, 2, 3]
3284 math_ops.reduce_sum(math_ops.abs(pixel_dif1), axis=sum_axis) +
3285 math_ops.reduce_sum(math_ops.abs(pixel_dif2), axis=sum_axis))
|
| D | image_ops_test.py | 4689 sum_axis = None
4696 sum_axis = (1, 2, 3)
4701 tot_var = np.sum(np.abs(dif1), axis=sum_axis) + \
4702 np.sum(np.abs(dif2), axis=sum_axis)
|