license import { backend_util , DataType , TypedArray , util } from '@tensorflow/tfjs-core' ; import { maxPoolPositions , pool } from '../utils/pool_utils' ; export function maxPoolWithArgmaxImpl ( xValues , xShape , dtype , includeBatchInIndex , convInfo ) { const strides = util . computeStrides ( xShape ) ; const maxPools = pool ( xValues , xShape , dtype , strides , convInfo , 'max' ) ; const maxPositions = maxPoolPositions ( xValues , xShape , dtype , convInfo , true , includeBatchInIndex ) ; return [ maxPools . values , maxPositions . values ] ; }