Searched refs:tile_spatial_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deep_conv2d.cc | 211 const int64 tile_spatial_size = tile_rows * tile_cols; in operator ()() local
214 ConstMatrixMap A(transform_matrix, tile_spatial_size, in operator ()()
217 MatrixMap C(out_buffer, tile_spatial_size, in_stride); in operator ()()
237 for (int64 i = 0; i < tile_spatial_size; ++i) { in operator ()()
386 const int64 tile_spatial_size = tile_rows * tile_cols; in operator ()() local
402 tile_spatial_size * base_filter_spatial_size; in operator ()()
412 tile_spatial_size * filter_shards_total * in_depth; in operator ()()
430 TensorShape({tile_spatial_size, base_filter_spatial_size}), in operator ()()
434 tile_spatial_size, base_filter_spatial_size, transform_matrix); in operator ()()
438 &filter_shards_col, &tile_spatial_size, &filter_in, in operator ()()
[all …]
|