Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op.cc969 const int64 input_size_per_batch = grad_in.NumElements() / batch_size; in launch() local
982 const int input_start = start * input_size_per_batch; in launch()
983 const int input_end = limit * input_size_per_batch; in launch()
987 const int64 cur_batch = index / input_size_per_batch; in launch()