Searched refs:in_shard (Results 1 – 3 of 3) sorted by relevance
383 static bool in_shard() { in in_shard() function390 if (in_shard() && in push_src()1273 if (!in_shard()) { in gather_tests()
385 static bool in_shard() { in in_shard() function392 if (in_shard() && in push_src()1365 if (!in_shard()) { in gather_tests()
177 EigenMatrixMap in_shard(input_backprop_flat.data() + in_start, 1, in SpatialMaxPoolWithArgMaxHelper() local179 in_shard.setConstant(T(0)); in SpatialMaxPoolWithArgMaxHelper()