Home
last modified time | relevance | path

Searched defs:Conv2DState (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddeep_conv2d.cc840 struct Conv2DState { struct
841 Conv2DState(const int64 tile_spatial_size, const int64 filter_shards_row, in Conv2DState() function
857 const int64 tile_spatial_size;
858 const int64 filter_shards_row;
859 const int64 filter_shards_col;
860 const T* input;
861 const T* tile_transform_matrix;
862 const T* output_transform_matrix;
863 T* buffer1;
864 T* buffer2;
[all …]