Searched refs:axes_tensor (Results 1 – 1 of 1) sorted by relevance
1640 const Tensor* axes_tensor = c->input_tensor(3); in SparseReduceShapeFn() local1641 if (shape_tensor != nullptr && axes_tensor != nullptr) { in SparseReduceShapeFn()1643 auto axes_vec = axes_tensor->flat<int32>(); in SparseReduceShapeFn()