Searched refs:scratch3_tensor (Results 1 – 1 of 1) sorted by relevance
545 Tensor scratch3_tensor; in operator ()() local549 scratch3_tensor.CopyFrom(*x_backprop_output, {rest_size, depth}), in operator ()()554 &scratch3_tensor)); in operator ()()559 typename TTypes<U, 2>::Tensor scratch3(scratch3_tensor.tensor<U, 2>()); in operator ()()599 redux_sum_u(d, rest_by_depth, scratch3_tensor, &scratch2_vec); in operator ()()