Searched refs:tile_transform_matrix (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | deep_conv2d.cc | 857 const T* tile_transform_matrix, const T* output_transform_matrix, in Conv2DState()
864 tile_transform_matrix(tile_transform_matrix), in Conv2DState()
875 const T* tile_transform_matrix; member
901 cs.tile_transform_matrix, cs.buffer1, cs.buffer2); in operator ()()
1003 T* tile_transform_matrix = in operator ()() local
1006 tile_transform_matrix); in operator ()()
1022 &tile_transform_matrix, &output_transform_matrix, in operator ()()
1124 filter_shards_col, input, tile_transform_matrix, in operator ()()
|