Searched refs:axes_vec (Results 1 – 2 of 2) sorted by relevance
2515 auto axes_vec = axes_tensor->flat<int32>(); in SparseReduceShapeFn() local2522 for (int i = 0; i < axes_vec.size(); i++) { in SparseReduceShapeFn()2523 axes.insert((axes_vec(i) + ndims) % ndims); in SparseReduceShapeFn()
559 vector axes_vec = %axes.as_constant();562 if (axes_vec.size() == 0) {575 for (int32 i = 0; i < axes_vec.size(); i++) {576 int32 axis = positive_axis(axes_vec[i], %input.rank);