Searched refs:input_backprop_index (Results 1 – 2 of 2) sorted by relevance
349 int input_backprop_index = out_arg_max_flat(index); in Compute() local351 CHECK(input_backprop_index >= 0 && in Compute()352 input_backprop_index < num_total_inputs) in Compute()353 << "Invalid input backprop index: " << input_backprop_index << ", " in Compute()355 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in Compute()
186 int input_backprop_index = out_arg_max_flat(index); in SpatialMaxPoolWithArgMaxHelper() local192 FastBoundsCheck(input_backprop_index - in_start, in_end - in_start); in SpatialMaxPoolWithArgMaxHelper()193 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in SpatialMaxPoolWithArgMaxHelper()