Searched defs:dim_lengths (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 182 std::vector<int64_t> dim_lengths{static_cast<int64_t>(operand.size())}; in ReduceWindow1DAdd() local
191 std::vector<int64_t> dim_lengths{operand.n1(), operand.n2(), operand.n3()}; in ReduceWindow3DAdd() local
238 std::vector<int64_t> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
251 std::vector<int64_t> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in ReduceWindow4DGeneric() local
334 std::vector<int64_t> dim_lengths{operand.n1(), operand.n2(), operand.n3(), in SelectAndScatter4DGePlus() local
|