Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc165 std::vector<int64> window_counts(window.size(), 0); in ReduceWindow1DGeneric() local
169 window_counts[i] = in ReduceWindow1DGeneric()
173 auto result = absl::make_unique<std::vector<float>>(window_counts[0]); in ReduceWindow1DGeneric()
176 for (int64 i0 = 0; i0 < window_counts[0]; ++i0) { in ReduceWindow1DGeneric()
210 std::vector<int64> window_counts(window.size(), 0); in ReduceWindow2DGeneric() local
214 window_counts[i] = in ReduceWindow2DGeneric()
219 absl::make_unique<Array2D<float>>(window_counts[0], window_counts[1]); in ReduceWindow2DGeneric()
222 for (int64 i0 = 0; i0 < window_counts[0]; ++i0) { in ReduceWindow2DGeneric()
223 for (int64 i1 = 0; i1 < window_counts[1]; ++i1) { in ReduceWindow2DGeneric()
259 std::vector<int64> window_counts(window.size(), 0); in ReduceWindow3DAdd() local
[all …]