Searched refs:out_arg_max_flat (Results 1 – 2 of 2) sorted by relevance
344 auto out_arg_max_flat = tensor_out_arg_max.flat<int64>(); in Compute() local349 int input_backprop_index = out_arg_max_flat(index); in Compute()
170 auto out_arg_max_flat = output_arg_max->flat<int64>(); in SpatialMaxPoolWithArgMaxHelper() local186 int input_backprop_index = out_arg_max_flat(index); in SpatialMaxPoolWithArgMaxHelper()