Searched defs:indices_bdim (Results 1 – 1 of 1) sorted by relevance
25 const Tensor& indices, std::optional<int64_t> indices_bdim, in embedding_batch_rule()56 const Tensor& indices_, std::optional<int64_t> indices_bdim, in embedding_dense_backward_batch_rule()