Searched refs:out_arg_max_ref (Results 1 – 2 of 2) sorted by relevance
310 int64& out_arg_max_ref = in Compute() local313 out_arg_max_ref == kInvalidMaxPoolingIndex) { in Compute()316 out_arg_max_ref = input_offset; in Compute()
151 int64& out_arg_max_ref = out_arg_max_mat.coeffRef(d, out_index); in SpatialMaxPoolWithArgMaxHelper() local153 out_arg_max_ref == kInvalidMaxPoolingIndex) { in SpatialMaxPoolWithArgMaxHelper()156 out_arg_max_ref = in_index * depth + d; in SpatialMaxPoolWithArgMaxHelper()158 out_arg_max_ref = (h * in_cols + w) * depth + d; in SpatialMaxPoolWithArgMaxHelper()