Searched refs:window_dimension (Results 1 – 7 of 7) sorted by relevance
87 for (const auto& window_dimension : window.dimensions()) { in ToString() local88 StrAppend(&str, prefix, format(window_dimension)); in ToString()207 bool IsTrivialWindowDimension(const WindowDimension& window_dimension) { in IsTrivialWindowDimension() argument208 return window_dimension.size() == 1 && window_dimension.stride() == 1 && in IsTrivialWindowDimension()209 window_dimension.padding_low() == 0 && in IsTrivialWindowDimension()210 window_dimension.padding_high() == 0 && in IsTrivialWindowDimension()211 window_dimension.window_dilation() == 1 && in IsTrivialWindowDimension()212 window_dimension.base_dilation() == 1; in IsTrivialWindowDimension()
67 bool IsTrivialWindowDimension(const WindowDimension& window_dimension);
56 int64 window_dimension = window_dimensions[i]; in MakePadding() local126 window_dimension - input_dimension, in MakePadding()
28 int64 window_dimension, in ComputePadding() argument30 return MakePadding({input_dimension}, {window_dimension}, {window_stride}, in ComputePadding()
357 const WindowDimension& window_dimension = in HandleReduceWindow() local360 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleReduceWindow()378 const WindowDimension& window_dimension = in HandleSelectAndScatter() local381 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleSelectAndScatter()
1873 for (const auto& window_dimension : window.dimensions()) { in HandleSelectAndScatter() local1874 window_dimension_sizes.push_back(window_dimension.size()); in HandleSelectAndScatter()1989 for (const auto& window_dimension : window.dimensions()) { in HandleReduceWindow() local1990 window_dimension_sizes.push_back(window_dimension.size()); in HandleReduceWindow()
3240 const WindowDimension& window_dimension = window.dimensions(i); in HandleReduceWindow() local3244 window_dimension.size() < pad_dimension.edge_padding_low() + 1) { in HandleReduceWindow()3251 window_dimension.size() != 1) { in HandleReduceWindow()