Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op.cc83 EigenIndexMatrixMap out_arg_max_mat( in SpatialMaxPoolWithArgMaxHelper() local
100 auto shard = [&params, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop, in SpatialMaxPoolWithArgMaxHelper()
122 out_arg_max_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
151 int64& out_arg_max_ref = out_arg_max_mat.coeffRef(d, out_index); in SpatialMaxPoolWithArgMaxHelper()