Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.h70 bool IsTrivialWindowDimension(const WindowDimension& window_dimension);
Dwindow_util.cc217 bool IsTrivialWindowDimension(const WindowDimension& window_dimension) { in IsTrivialWindowDimension() function
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc780 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleDynamicWindowSamePadding()
1245 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleReduceWindow()
Ddynamic_padder.cc1194 if (window_util::IsTrivialWindowDimension(window_dim)) { in RewriteDynamicReduceWindowSamePadding()
1231 if (window_util::IsTrivialWindowDimension(window_dim)) { in RewriteDynamicSelectAndScatterSamePadding()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc232 if (window_util::IsTrivialWindowDimension(wd)) { in GetShardingReplicatedOnWindowedDimension()
707 window_util::IsTrivialWindowDimension(window.dimensions(i)); in ReshardAsWindowedInput()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2677 !window_util::IsTrivialWindowDimension(window.dimensions(i)) && in ReduceWindow()
3277 !window_util::IsTrivialWindowDimension(window.dimensions(i)) && in SelectAndScatter()