Searched refs:output_arg_max (Results 1 – 1 of 1) sorted by relevance
59 OpKernelContext* context, Tensor* output, Tensor* output_arg_max, in SpatialMaxPoolWithArgMaxHelper() argument84 output_arg_max->flat<int64>().data(), params.depth, in SpatialMaxPoolWithArgMaxHelper()101 &output_arg_max, &out_backprop, in SpatialMaxPoolWithArgMaxHelper()170 auto out_arg_max_flat = output_arg_max->flat<int64>(); in SpatialMaxPoolWithArgMaxHelper()