Searched refs:out_arg_max_shard (Results 1 – 1 of 1) sorted by relevance
121 EigenIndexMatrixMap out_arg_max_shard( in SpatialMaxPoolWithArgMaxHelper() local124 out_arg_max_shard.setConstant(kInvalidMaxPoolingIndex); in SpatialMaxPoolWithArgMaxHelper()