Searched refs:indices_op (Results 1 – 2 of 2) sorted by relevance
51 values_op, indices_op = nn_ops.top_k(inputs, k, sorted=sorted)52 values, indices = self.evaluate([values_op, indices_op])55 self.assertShapeEqual(np_expected_indices, indices_op)
246 auto indices_op = indices.getDefiningOp(); in GetFirstIfIndicesAreContiguous() local247 if (!indices_op) return -1; in GetFirstIfIndicesAreContiguous()248 auto const_op = llvm::dyn_cast<TF::ConstOp>(indices_op); in GetFirstIfIndicesAreContiguous()