Searched refs:sparse_dims (Results 1 – 1 of 1) sorted by relevance
235 const Tensor& sparse_dims = context->input(1); in Compute() local241 const TensorShape& sparse_dims_shape = sparse_dims.shape(); in Compute()242 const auto& axes_sparse_flat = sparse_dims.flat<Tidx>(); in Compute()246 sparse_dims.dims())); in Compute()