Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dfractional_max_pool_op.cc342 auto out_backprop_flat = out_backprop.flat<T>(); in Compute() local
345 int num_total_outputs = out_backprop_flat.size(); in Compute()
355 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in Compute()
Dmaxpooling_op.cc171 auto out_backprop_flat = out_backprop.flat<T>(); in SpatialMaxPoolWithArgMaxHelper() local
193 input_backprop_flat(input_backprop_index) += out_backprop_flat(index); in SpatialMaxPoolWithArgMaxHelper()