Searched defs:output_arg_max (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op.cc | 57 OpKernelContext* context, Tensor* output, Tensor* output_arg_max,
74 output_arg_max->flat<int64>().data(), params.depth,
91 &output_arg_max, &out_backprop](int64 start, int64 limit) {
156 auto out_arg_max_flat = output_arg_max->flat<int64>();
56 SpatialMaxPoolWithArgMaxHelper( OpKernelContext* context, Tensor* output, Tensor* output_arg_max, Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, const PoolParameters& params, const Padding& padding) argument
|
Completed in 147 milliseconds