Searched defs: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
196 std::vector<int64> dim_lengths{static_cast<int64>(operand.size())}; in ReduceWindow1DAdd() local
208 std::vector<int64> dim_lengths{operand.height(), operand.width()}; in ReduceWindow2DGeneric() local
247 std::vector<int64> dim_lengths{operand.height(), operand.width()}; in ReduceWindow2DAdd() local
256 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3()}; in ReduceWindow3DAdd() local
303 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
316 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
398 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in SelectAndScatter4DGePlus() local
|