Searched refs:out_arg_max_mat (Results 1 – 1 of 1) sorted by relevance
83 EigenIndexMatrixMap out_arg_max_mat( in SpatialMaxPoolWithArgMaxHelper() local100 auto shard = [¶ms, &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()