Home
last modified time | relevance | path

Searched refs:out_tile_cols (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddeep_conv2d.cc49 int out_tile_rows, int out_tile_cols, int in_depth, in GetDeepConvCost() argument
60 const int64 output_tile_spatial_size = out_tile_rows * out_tile_cols; in GetDeepConvCost()
66 const int64 col_tiles = (out_cols + out_tile_cols - 1) / out_tile_cols; in GetDeepConvCost()
759 const int64 out_tile_cols = transform->output_shape().cols; in operator ()() local
760 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()()
811 for (int64 ot_col = 0; ot_col < out_tile_cols; ++ot_col) { in operator ()()
816 const int64 out_buf_index = ot_row * out_tile_cols + ot_col; in operator ()()
951 const int64 out_tile_cols = transform->output_shape().cols; in operator ()() local
952 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()()
1006 out_depth, out_tile_rows, out_tile_cols, filter_shards_row, in operator ()()
[all …]