Searched refs:dim_lengths (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 181 std::vector<int64> dim_lengths{static_cast<int64>(operand.size())}; in ReduceWindow1DAdd() local
184 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow1DAdd()
190 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3()}; in ReduceWindow3DAdd() local
191 auto padding_both = xla::MakePadding(dim_lengths, window, stride, padding); in ReduceWindow3DAdd()
197 WindowCount(dim_lengths[i], window[i], stride[i], padding); in ReduceWindow3DAdd()
237 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
241 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow4DGeneric()
250 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
256 int64 padded_width = padding[i].first + dim_lengths[i] + padding[i].second; in ReduceWindow4DGeneric()
332 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in SelectAndScatter4DGePlus() local
[all …]
|