Searched defs:num_index_dims (Results 1 – 6 of 6) sorted by relevance
52 int64 num_index_dims; in XlaGather() local281 const int64 num_index_dims = in Compile() local
43 const int64 num_index_dims = indices_shape.dim_size(indices_shape.dims() - 1); in ValidateUpdateShape() local
47 int64 num_index_dims = 1; in XlaScatter() local
124 const int64 num_index_dims = 2; in GetMatrixDiagonalViaGather() local
2233 int64_t num_index_dims = indices_ty.getShape().back(); in Verify() local
3931 int64_t num_index_dims = indices_ty.getShape().back(); in matchAndRewrite() local