Searched defs: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
190 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3()}; in ReduceWindow3DAdd() local
237 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
250 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
332 std::vector<int64> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in SelectAndScatter4DGePlus() local
|