Searched refs:scratch3_tensor (Results 1 – 1 of 1) sorted by relevance
560 Tensor scratch3_tensor; in operator ()() local564 scratch3_tensor.CopyFrom(*x_backprop_output, {rest_size, depth}), in operator ()()569 &scratch3_tensor)); in operator ()()574 typename TTypes<U, 2>::Tensor scratch3(scratch3_tensor.tensor<U, 2>()); in operator ()()609 redux_sum_u(d, rest_by_depth, scratch3_tensor, &scratch2_vec); in operator ()()