Searched refs:axes_vec (Results 1 – 1 of 1) sorted by relevance
1643 auto axes_vec = axes_tensor->flat<int32>(); in SparseReduceShapeFn() local1647 for (int i = 0; i < axes_vec.size(); i++) { in SparseReduceShapeFn()1648 axes.insert((axes_vec(i) + ndims) % ndims); in SparseReduceShapeFn()