Searched refs:window_dimension (Results 1 – 5 of 5) sorted by relevance
57 int64 window_dimension = window_dimensions[i]; in MakePadding() local127 window_dimension - input_dimension, in MakePadding()
28 int64 window_dimension, in ComputePadding() argument30 return MakePadding({input_dimension}, {window_dimension}, {window_stride}, in ComputePadding()
87 for (const auto& window_dimension : window.dimensions()) { in ToString() local88 StrAppend(&str, prefix, format(window_dimension)); in ToString()
1543 for (const auto& window_dimension : window.dimensions()) { in HandleSelectAndScatter() local1544 window_dimension_sizes.push_back(window_dimension.size()); in HandleSelectAndScatter()1659 for (const auto& window_dimension : window.dimensions()) { in HandleReduceWindow() local1660 window_dimension_sizes.push_back(window_dimension.size()); in HandleReduceWindow()
1772 const WindowDimension& window_dimension = window.dimensions(i); in HandleReduceWindow() local1776 window_dimension.size() < pad_dimension.edge_padding_low() + 1) { in HandleReduceWindow()1783 window_dimension.size() != 1) { in HandleReduceWindow()