Searched refs:dim_lengths (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 164 std::vector<int64> dim_lengths{static_cast<int64>(operand.size())}; in ReduceWindow1DGeneric() local
168 int64 padded_width = padding[i].first + dim_lengths[i] + padding[i].second; in ReduceWindow1DGeneric()
181 if (i0_base + i0_win >= 0 && i0_base + i0_win < dim_lengths[0]) { in ReduceWindow1DGeneric()
196 std::vector<int64> dim_lengths{static_cast<int64>(operand.size())}; in ReduceWindow1DAdd() local
199 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow1DAdd()
208 std::vector<int64> dim_lengths{operand.height(), operand.width()}; in ReduceWindow2DGeneric() local
213 int64 padded_width = padding[i].first + dim_lengths[i] + padding[i].second; in ReduceWindow2DGeneric()
247 std::vector<int64> dim_lengths{operand.height(), operand.width()}; in ReduceWindow2DAdd() local
250 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow2DAdd()
256 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3()}; in ReduceWindow3DAdd() local
[all …]
|