Searched refs:out_backprop_flat (Results 1 – 2 of 2) sorted by relevance
342 auto out_backprop_flat = out_backprop.flat<T>(); in Compute() local345 int num_total_outputs = out_backprop_flat.size(); in Compute()355 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in Compute()
171 auto out_backprop_flat = out_backprop.flat<T>(); in SpatialMaxPoolWithArgMaxHelper() local193 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in SpatialMaxPoolWithArgMaxHelper()