Home
last modified time | relevance | path

Searched refs:input_backprop_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfractional_max_pool_op.cc349 int input_backprop_index = out_arg_max_flat(index); in Compute() local
351 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()
Dmaxpooling_op.cc186 int input_backprop_index = out_arg_max_flat(index); in SpatialMaxPoolWithArgMaxHelper() local
192 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()