Searched defs:is_batched (Results 1 – 7 of 7) sorted by relevance
901 bool is_batched = input.dim() == 2; in forward() local958 bool is_batched = input.dim() == 2; in forward() local1009 bool is_batched = input.dim() == 2; in forward() local
101 bool is_batched = (targ_dim > 0); in forward() local
64 auto is_batched = inp_dim == batched_dim; in _dropoutNd_helper() local
228 def _composite_to_tensors(value, is_batched=False): argument
499 auto is_batched = log_probs_.dim() == 3; in ctc_loss_impl() local
766 const auto is_batched = (input.dim() == dim_count_batch); in batchify() local
1052 auto is_batched = n_batch_dim > 0; in dense_to_sparse_compressed() local