Searched refs:axes_vec (Results 1 – 2 of 2) sorted by relevance
2423 auto axes_vec = axes_tensor->flat<int32>(); in SparseReduceShapeFn() local2427 for (int i = 0; i < axes_vec.size(); i++) { in SparseReduceShapeFn()2428 axes.insert((axes_vec(i) + ndims) % ndims); in SparseReduceShapeFn()
583 vector axes_vec = %axes.as_constant();586 if (axes_vec.size() == 0) {594 for (int i = 0; i < axes_vec.size(); i++) {595 int32_t axis = positive_axis(axes_vec[i], %input.rank);