Searched refs:sparse_tensor_shape (Results 1 – 1 of 1) sorted by relevance
86 const VarDimArray& sparse_tensor_shape) { in CheckGroup() argument100 const auto expected_rank = sparse_tensor_shape.size(); in CheckGroup()105 const auto dim_size = sparse_tensor_shape[j]; in CheckGroup()227 const VarDimArray& sparse_tensor_shape, in PopulateFromSparseGroup() argument229 CheckGroup<T>(ctx, group, sparse_tensor_shape); in PopulateFromSparseGroup()