Searched defs:Conv2DState (Results 1 – 1 of 1) sorted by relevance
854 struct Conv2DState { struct855 Conv2DState(const int64_t tile_spatial_size, const int64_t filter_shards_row, in Conv2DState() argument871 const int64_t tile_spatial_size;872 const int64_t filter_shards_row;873 const int64_t filter_shards_col;874 const T* input;875 const T* tile_transform_matrix;876 const T* output_transform_matrix;877 T* buffer1;878 T* buffer2;[all …]