HomeSort by relevance Sort by last modified time
    Searched refs:out_arg_max_flat (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
fractional_max_pool_op.cc 344 auto out_arg_max_flat = tensor_out_arg_max.flat<int64>(); variable
349 int input_backprop_index = out_arg_max_flat(index);
maxpooling_op.cc 156 auto out_arg_max_flat = output_arg_max->flat<int64>();
172 int input_backprop_index = out_arg_max_flat(index);
    [all...]

Completed in 354 milliseconds