Searched refs:Conv2DState (Results 1 – 1 of 1) sorted by relevance
840 struct Conv2DState { struct841 Conv2DState(const int64 tile_spatial_size, const int64 filter_shards_row, in Conv2DState() argument881 const Conv2DState<T>& cs, const int64 in_r, const int64 in_c, in operator ()()1105 Conv2DState<T> conv_state(tile_spatial_size, filter_shards_row, in operator ()()