Searched refs:redux_sum_u (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 430 const functor::ReduceOuterDimensions<U, U, U, ScalarSum> redux_sum_u; in operator ()() local
471 redux_sum_u(d, rest_by_depth, scratch_rest_by_depth, scale_backprop_output); in operator ()()
492 redux_sum_u(d, rest_by_depth, scratch_rest_by_depth, &scratch_one_by_depth); in operator ()()
575 const functor::ReduceOuterDimensions<U, U, U, ScalarSum> redux_sum_u; in operator ()() local
599 redux_sum_u(d, rest_by_depth, scratch3_tensor, &scratch2_vec); in operator ()()
|