Searched defs:include_batch_in_index (Results 1 – 3 of 3) sorted by relevance
71 dtype* top_data, int64* mask, const bool include_batch_in_index) { in MaxPoolForwardNCHW()145 dtype* top_data, int64* mask, const bool include_batch_in_index) { in MaxPoolForwardNHWC()243 const bool include_batch_in_index) { in MaxPoolBackward()370 const bool include_batch_in_index) { in MaxPoolGradBackward()409 const bool include_batch_in_index) { in operator ()()460 const bool include_batch_in_index) { in operator ()()505 const bool include_batch_in_index) { in operator ()()
61 const PoolParameters& params, const bool include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper()
3834 include_batch_in_index=False, argument3905 include_batch_in_index=False): argument