Searched defs:include_batch_in_index (Results 1 – 4 of 4) sorted by relevance
73 const bool include_batch_in_index) { in MaxPoolForwardNCHW()150 const bool include_batch_in_index) { in MaxPoolForwardNHWC()208 const bool include_batch_in_index) { in MaxPoolBackward()338 const bool include_batch_in_index) { in MaxPoolGradBackward()380 const bool include_batch_in_index) { in operator ()()409 const bool include_batch_in_index) { in operator ()()458 const bool include_batch_in_index) { in operator ()()
65 const PoolParameters& params, const bool include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper()
109 bool include_batch_in_index; member
4777 include_batch_in_index=False, argument4849 include_batch_in_index=False): argument