Searched refs:input_backprop_index (Results 1 – 3 of 3) sorted by relevance
374 int input_backprop_index = out_arg_max_flat(index); in Compute() local377 input_backprop_index >= 0 && input_backprop_index < num_total_inputs, in Compute()379 "Invalid input backprop index: ", input_backprop_index, ", ", in Compute()381 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in Compute()
197 int input_backprop_index = out_arg_max_flat(index); in SpatialMaxPoolWithArgMaxHelper() local203 FastBoundsCheck(input_backprop_index - in_start, in_end - in_start); in SpatialMaxPoolWithArgMaxHelper()205 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in SpatialMaxPoolWithArgMaxHelper()
32 int input_backprop_index = out_arg_max_flat(index);33 FastBoundsCheck(input_backprop_index - in_start, in_end - in_start);34 input_backprop_flat(input_backprop_index) += out_backprop_flat(index);