Searched defs:n_tensors (Results 1 – 9 of 9) sorted by relevance
84 int64_t n_tensors = self.size(0); in chunk_nested_tensor() local140 int64_t n_tensors = self.size(0); in split_with_sizes_nested() local
42 size_t n_tensors = params.size(); in _fused_adam_kernel_cpu_() local118 size_t n_tensors = params.size(); in _fused_adamw_kernel_cpu_() local
36 size_t n_tensors = params.size(); in _fused_adagrad_kernel_cpu_() local
38 size_t n_tensors = params.size(); in _fused_sgd_kernel_cpu_() local
134 const size_t n_tensors = tensor_lists[0].size(); in multi_tensor_apply() local225 const size_t n_tensors = tensor_lists[0].size(); in multi_tensor_apply() local
167 const int64_t n_tensors = param.size(0); in check_for_seq_len_0_and_consistent_head_dim_nested_tensor_helper() local
91 const int64_t n_tensors = tensor_strides.size(0); in is_safe_to_get_storage_as_tensor() local
743 const int64_t n_tensors = params.query.size(0); in check_for_seq_len_1_nested_tensor() local
168 size_t n_scalars = 0, n_tensors = 0; in lint_python() local