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 ()()
66 const PoolParameters& params, const bool include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper()
112 bool include_batch_in_index; member
5131 include_batch_in_index=False, argument5206 include_batch_in_index=False): argument