Home
last modified time | relevance | path

Searched refs:out_arg_max_flat (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfractional_max_pool_op.cc344 auto out_arg_max_flat = tensor_out_arg_max.flat<int64>(); in Compute() local
349 int input_backprop_index = out_arg_max_flat(index); in Compute()
Dmaxpooling_op.cc170 auto out_arg_max_flat = output_arg_max->flat<int64>(); in SpatialMaxPoolWithArgMaxHelper() local
186 int input_backprop_index = out_arg_max_flat(index); in SpatialMaxPoolWithArgMaxHelper()