Searched refs:num_backward_tokens (Results 1 – 5 of 5) sorted by relevance
271 num_backward_tokens = fw_metadata.num_backward_tokens367 if bw_module is not None and num_backward_tokens > 0:382 do(bw_module, "backward", num_backward_tokens)387 fw_metadata.num_backward_tokens = 0
388 num_tokens > 0 or fw_metadata.num_backward_tokens > 0493 num_backward_tokens: int = inner_meta.num_backward_tokens498 + num_backward_tokens501 if (inner_meta.num_outputs_rng_offset + num_backward_tokens) > 0:503 : -(inner_meta.num_outputs_rng_offset + num_backward_tokens)
356 num_backward_tokens: int = 0 variable in ViewAndMutationMeta573 and self.num_backward_tokens == other.num_backward_tokens
707 meta.num_backward_tokens = len(bwd_out_tokens)
1733 bw_tokens = [None] * CompiledFunction.metadata.num_backward_tokens1988 num_bw_tokens = CompiledFunction.metadata.num_backward_tokens