| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | custom_lstms.py | 38 num_layers, argument
74 num_layers, argument
285 def init_stacked_lstm(num_layers, layer, first_layer_args, other_layer_args): argument
295 def __init__(self, num_layers, layer, first_layer_args, other_layer_args): argument
325 def __init__(self, num_layers, layer, first_layer_args, other_layer_args): argument
353 def __init__(self, num_layers, layer, first_layer_args, other_layer_args): argument
423 def test_script_stacked_rnn(seq_len, batch, input_size, hidden_size, num_layers): argument
449 def test_script_stacked_bidir_rnn(seq_len, batch, input_size, hidden_size, num_layers): argument
481 seq_len, batch, input_size, hidden_size, num_layers argument
494 def test_script_stacked_lnlstm(seq_len, batch, input_size, hidden_size, num_layers): argument
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/iris/ |
| H A D | iris_resolve.c | 138 unsigned num_layers = in resolve_image_views() local
212 unsigned num_layers = in iris_predraw_resolve_framebuffer() local
312 unsigned num_layers = in iris_postdraw_update_image_resolve_tracking() local
351 unsigned num_layers = in iris_postdraw_update_resolve_tracking() local
384 unsigned num_layers = in iris_postdraw_update_resolve_tracking() local
572 uint32_t num_layers, in iris_mcs_exec()
678 unsigned int num_layers, enum isl_aux_op op) in iris_hiz_exec()
833 uint32_t start_layer, uint32_t num_layers) in miptree_layer_range_length()
851 unsigned start_layer, unsigned num_layers) in iris_has_invalid_primary()
878 uint32_t start_layer, uint32_t num_layers, in iris_resource_prepare_access()
[all …]
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/crocus/ |
| H A D | crocus_resolve.c | 150 unsigned num_layers = in resolve_image_views() local
235 unsigned num_layers = in crocus_predraw_resolve_framebuffer() local
332 unsigned num_layers = in crocus_postdraw_update_resolve_tracking() local
380 unsigned num_layers = in crocus_postdraw_update_resolve_tracking() local
572 uint32_t num_layers) in crocus_mcs_partial_resolve()
607 unsigned int num_layers, enum isl_aux_op op) in crocus_hiz_exec()
772 uint32_t start_layer, uint32_t num_layers) in miptree_layer_range_length()
790 unsigned start_layer, unsigned num_layers) in crocus_has_invalid_primary()
820 uint32_t start_layer, uint32_t num_layers, in crocus_resource_prepare_access()
878 uint32_t start_layer, uint32_t num_layers, in crocus_resource_finish_write()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Gru.cpp | 47 int64_t num_layers, in gru_input()
141 int64_t num_layers) { in pack_linear_op_contexts()
190 int64_t num_layers, in GruPackedContext()
232 const int64_t num_layers = get_val(Packed::NumLayers).toInt(); in unpack() local
265 int64_t num_layers, in create_gru_context()
290 const int64_t num_layers = in run_gru_context() local
|
| H A D | Lstm.cpp | 56 int64_t num_layers, in lstm_input()
177 int64_t num_layers) { in pack_lstm_linear_op_contexts()
232 int64_t num_layers, in LstmPackedContext()
274 const int64_t num_layers = get_val(Packed::NumLayers).toInt(); in unpack() local
307 int64_t num_layers, in create_lstm_context()
336 const int64_t num_layers = in run_lstm_context() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | transformer.cpp | 20 int64_t num_layers) in TransformerEncoderOptions()
25 int64_t num_layers) in TransformerEncoderOptions()
30 int64_t num_layers) in TransformerDecoderOptions()
35 int64_t num_layers) in TransformerDecoderOptions()
|
| /aosp_15_r20/external/mesa3d/src/intel/blorp/ |
| H A D | blorp_clear.c | 403 uint32_t level, uint32_t start_layer, uint32_t num_layers, in blorp_fast_clear()
514 uint32_t level, uint32_t start_layer, uint32_t num_layers, in blorp_clear()
726 uint32_t num_layers, in blorp_clear_stencil_as_rgba()
831 uint32_t num_layers, in blorp_clear_depth_stencil()
947 uint32_t start_layer, uint32_t num_layers, in blorp_hiz_clear_depth_stencil()
1051 uint32_t start_layer, uint32_t num_layers, in blorp_clear_attachments()
1115 uint32_t start_layer, uint32_t num_layers, in blorp_ccs_resolve()
1306 uint32_t start_layer, uint32_t num_layers) in blorp_mcs_partial_resolve()
1376 uint32_t start_layer, uint32_t num_layers) in blorp_mcs_ambiguate()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | RNN.cpp | 39 int64_t num_layers, in mkldnn_rnn_layer()
64 int64_t num_layers, in mkldnn_rnn_layer_backward()
92 int64_t num_layers; member
227 int64_t num_layers, in mkldnn_rnn_layer()
311 int64_t num_layers, in mkldnn_rnn_layer_backward()
458 int64_t num_layers, bool has_biases, bool batch_first, double dropout_p, in mkldnn_rnn()
538 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in mkldnn_impl()
553 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in lstm_mkldnn()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | cudnn_rnn_ops_test.cc | 45 int num_layers = 5; in TEST() local
82 int num_layers = 5; in TEST() local
119 int num_layers = 5; in TEST() local
161 int num_layers = 5; in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | transformer.cpp | 245 size_t num_layers = layers->size(); in reset_parameters() local
266 size_t num_layers = layers->size(); in forward() local
312 size_t num_layers = layers->size(); in reset_parameters() local
336 size_t num_layers = layers->size(); in forward() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | gradients_test.py | 49 def __init__(self, activation_size, num_layers): argument
62 def fully_connected_model_fn(batch_size, activation_size, num_layers): argument
99 def create_fc_batch_jacobian(batch_size, activation_size, num_layers): argument
157 def create_fc_per_eg_grad(batch_size, activation_size, num_layers): argument
328 def create_fc_per_eg_jacobians(batch_size, activation_size, num_layers): argument
|
| /aosp_15_r20/external/libopus/dnn/torch/osce/models/ |
| H A D | fd_discriminator.py | 595 num_layers=5, argument
723 num_layers=5, argument
791 num_layers=5, argument
836 num_layers=5, argument
884 num_layers=5, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 1082 int64_t num_layers, double dropout_p, bool train) { in apply_layer_stack()
1112 int64_t num_layers, double dropout_p, bool train, bool bidirectional) { in _rnn_impl()
1129 int64_t num_layers, double dropout_p, bool train, bool bidirectional) { in _rnn_impl_with_concat()
1138 int64_t num_layers, double dropout_p, bool train, bool bidirectional) { in _lstm_impl()
1404 int64_t num_layers, in quantized_gru_data_legacy()
1433 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in lstm()
1494 int64_t num_layers, double dropout_p, bool train, bool bidirectional) { in lstm()
1693 int64_t num_layers, in quantized_lstm_input()
1733 int64_t num_layers, in quantized_lstm_input_legacy()
1753 int64_t num_layers, in quantized_lstm_data()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 42 …ptional<Tensor>& grad_cy_r_opt, int64_t mode, int64_t hidden_size, int64_t num_layers, bool batch_… in miopen_rnn_backward()
74 int64_t num_layers; member
336 int64_t num_layers = rnn.num_directions() * rnn.num_layers; in get_parameters() local
762 …ptional<Tensor>& grad_cy_r_opt, int64_t mode, int64_t hidden_size, int64_t num_layers, bool batch_… in miopen_rnn_backward()
825 int64_t num_layers, double dropout_p, bool train, bool bidirectional) { in _miopen_impl()
847 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in _miopen_impl()
900 int64_t num_layers, double dropout_p, bool train, bool bidirectional) { in lstm_packed_miopen()
|
| /aosp_15_r20/external/webrtc/modules/video_coding/svc/ |
| H A D | svc_rate_allocator.cc | 90 static std::vector<DataRate> SplitBitrate(size_t num_layers, in SplitBitrate()
408 const NumLayers num_layers = GetNumLayers(codec); in GetMaxBitrate() local
438 const NumLayers num_layers = GetNumLayers(codec); in GetLayerStartBitrates() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | transformercoder.h | 41 int64_t num_layers) in TransformerEncoderImpl()
102 int64_t num_layers) in TransformerDecoderImpl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 84 int64_t num_layers, in _cudnn_rnn_backward()
155 int64_t num_layers; member
715 int64_t num_layers = rnn.num_directions() * rnn.num_layers; in get_parameters() local
1246 int64_t num_layers, in copy_weights_to_flat_buf_views()
2102 int64_t num_layers, in _cudnn_rnn_backward()
2390 int64_t num_layers, in try_get_weight_buf()
2505 int64_t num_layers, in _cudnn_impl()
2578 int64_t num_layers, in _cudnn_impl()
2728 int64_t num_layers, in lstm_packed_cudnn()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/ |
| H A D | si_test_image_copy_region.c | 48 unsigned num_layers = util_num_layers(tex, level); in set_random_pixels() local
84 unsigned num_layers = util_num_layers(tex1, level); in set_random_pixels_for_2_textures() local
129 unsigned num_layers = util_num_layers(tex, level); in compare_textures() local
162 unsigned num_layers = util_num_layers(tex1, level); in compare_gpu_textures() local
|
| /aosp_15_r20/external/pytorch/functorch/examples/compilation/ |
| H A D | linear_train.py | 28 def __init__(self, num_layers=3, features=100): argument
41 num_layers = 8 variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | batch_norm_benchmark.py | 64 def build_graph(device, input_shape, axes, num_layers, mode, scale, train): argument
122 def _run_graph(self, device, input_shape, axes, num_layers, mode, scale, argument
|
| /aosp_15_r20/external/mesa3d/src/broadcom/vulkan/ |
| H A D | v3dv_meta_copy.c | 456 uint32_t num_layers; in copy_image_to_buffer_tlb() local
917 uint32_t num_layers; in copy_image_to_buffer_blit() local
1078 uint32_t num_layers; in copy_image_to_buffer_texel_buffer() local
1347 uint32_t num_layers; in copy_image_tlb() local
1934 uint32_t num_layers; in copy_buffer_to_image_tfu() local
2031 uint32_t num_layers; in copy_buffer_to_image_tlb() local
2638 uint32_t num_layers; in texel_buffer_shader_copy() local
2972 uint32_t num_layers; in copy_buffer_to_image_blit() local
4883 uint32_t num_layers; in resolve_image_tlb() local
|
| /aosp_15_r20/external/libaom/av1/encoder/ |
| H A D | ml.c | 39 const int num_layers = nn_config->num_hidden_layers; in av1_nn_predict_c() local
126 const int num_layers = nn_config->num_hidden_layers; in av1_nn_predict_v2() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/winsys/radeon/drm/ |
| H A D | radeon_drm_surface.c | 251 unsigned num_layers; in si_compute_cmask() local
264 struct radeon_surf *surf, unsigned num_layers) in si_compute_htile()
|
| /aosp_15_r20/external/mesa3d/src/mesa/state_tracker/ |
| H A D | st_draw_feedback.c | 276 unsigned num_layers = res->depth0; in st_feedback_draw_vbo() local
355 unsigned width, height, num_layers, row_stride, img_stride; in st_feedback_draw_vbo() local
|
| /aosp_15_r20/external/libva/va/ |
| H A D | va_drmcommon.h | 150 uint32_t num_layers; member
196 uint32_t num_layers; member
|