Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc312 int64 window_element_count = 1; in HandleReduceWindow() local
314 window_element_count *= dimension.size(); in HandleReduceWindow()
319 (window_element_count - 1) * output_element_count; in HandleReduceWindow()
342 int64 window_element_count = 1; in HandleSelectAndScatter() local
344 window_element_count *= dimension.size(); in HandleSelectAndScatter()
346 const int64 select_count = source_element_count * (window_element_count - 1); in HandleSelectAndScatter()