Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc87 for (const auto& window_dimension : window.dimensions()) { in ToString() local
88 StrAppend(&str, prefix, format(window_dimension)); in ToString()
207 bool IsTrivialWindowDimension(const WindowDimension& window_dimension) { in IsTrivialWindowDimension() argument
208 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()
Dwindow_util.h67 bool IsTrivialWindowDimension(const WindowDimension& window_dimension);
/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc56 int64 window_dimension = window_dimensions[i]; in MakePadding() local
126 window_dimension - input_dimension, in MakePadding()
Dpadding_test.cc28 int64 window_dimension, in ComputePadding() argument
30 return MakePadding({input_dimension}, {window_dimension}, {window_stride}, in ComputePadding()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc357 const WindowDimension& window_dimension = in HandleReduceWindow() local
360 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleReduceWindow()
378 const WindowDimension& window_dimension = in HandleSelectAndScatter() local
381 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleSelectAndScatter()
Dhlo_evaluator_typed_visitor.h1873 for (const auto& window_dimension : window.dimensions()) { in HandleSelectAndScatter() local
1874 window_dimension_sizes.push_back(window_dimension.size()); in HandleSelectAndScatter()
1989 for (const auto& window_dimension : window.dimensions()) { in HandleReduceWindow() local
1990 window_dimension_sizes.push_back(window_dimension.size()); in HandleReduceWindow()
Dalgebraic_simplifier.cc3240 const WindowDimension& window_dimension = window.dimensions(i); in HandleReduceWindow() local
3244 window_dimension.size() < pad_dimension.edge_padding_low() + 1) { in HandleReduceWindow()
3251 window_dimension.size() != 1) { in HandleReduceWindow()