Searched defs:num_index_dims (Results 1 – 6 of 6) sorted by relevance
52 int64_t num_index_dims; in XlaGather() local281 const int64_t num_index_dims = in Compile() local
43 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
2289 int64_t num_index_dims = indices_ty.getShape().back(); in Verify() local
1672 int64_t num_index_dims = indices_ty.getDimSize(indices_rank - 1); in matchAndRewrite() local4660 int64_t num_index_dims = indices_ty.getShape().back(); in matchAndRewrite() local