Searched refs:use_nested_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/ |
| H A D | transformer.cpp | 69 const bool use_nested_tensor) { in norm() argument
104 const bool use_nested_tensor = src.is_nested(); in transformer_encoder_layer_forward() local
107 … x = norm(x, embed_dim, layer_norm_eps, layer_norm_weight_1, layer_norm_bias_1, use_nested_tensor); in transformer_encoder_layer_forward()
126 … x = norm(x, embed_dim, layer_norm_eps, layer_norm_weight_1, layer_norm_bias_1, use_nested_tensor); in transformer_encoder_layer_forward()
133 … x = norm(x, embed_dim, layer_norm_eps, layer_norm_weight_2, layer_norm_bias_2, use_nested_tensor); in transformer_encoder_layer_forward()
145 … x = norm(x, embed_dim, layer_norm_eps, layer_norm_weight_2, layer_norm_bias_2, use_nested_tensor); in transformer_encoder_layer_forward()
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | transformer.py | 347 self.use_nested_tensor = enable_nested_tensor
382 self.use_nested_tensor = False
440 elif not self.use_nested_tensor:
|