Searched defs:num_index_dims (Results 1 – 6 of 6) sorted by relevance
53 int64_t num_index_dims; in XlaGather() local285 const int64_t num_index_dims = in Compile() local
44 const int64_t num_index_dims = in ValidateUpdateShape() local
47 int64_t num_index_dims = 1; in XlaScatter() local
125 const int64_t num_index_dims = 2; in GetMatrixDiagonalViaGather() local
2414 int64_t num_index_dims = indices_ty.getShape().back(); in verify() local
1496 int64_t num_index_dims = indices_ty.getDimSize(indices_rank - 1); in matchAndRewrite() local4461 int64_t num_index_dims = indices_ty.getShape().back(); in matchAndRewrite() local