Home
last modified time | relevance | path

Searched defs:num_layers (Results 1 – 25 of 68) sorted by relevance

123

/third_party/mindspore/tests/st/ops/gpu/
Dtest_lstm_op.py32 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
164 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
321 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
595 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
869 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
/third_party/mindspore/tests/st/ops/cpu/
Dtest_lstm_op.py39 num_layers=1, argument
102 …def __init__(self, batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, dropo… argument
205 …def __init__(self, batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, dropo… argument
293 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
/third_party/mesa3d/src/gallium/drivers/iris/
Diris_resolve.c138 unsigned num_layers = in resolve_image_views() local
199 unsigned num_layers = in iris_predraw_resolve_framebuffer() local
289 unsigned num_layers = in iris_postdraw_update_resolve_tracking() local
322 unsigned num_layers = in iris_postdraw_update_resolve_tracking() local
466 uint32_t num_layers) in iris_mcs_partial_resolve()
537 unsigned int num_layers, enum isl_aux_op op, in iris_hiz_exec()
672 uint32_t start_layer, uint32_t num_layers) in miptree_layer_range_length()
690 unsigned start_layer, unsigned num_layers) in iris_has_invalid_primary()
717 uint32_t start_layer, uint32_t num_layers, in iris_resource_prepare_access()
772 uint32_t start_layer, uint32_t num_layers, in iris_resource_finish_write()
[all …]
/third_party/mesa3d/src/gallium/drivers/crocus/
Dcrocus_resolve.c150 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()
775 uint32_t start_layer, uint32_t num_layers) in miptree_layer_range_length()
793 unsigned start_layer, unsigned num_layers) in crocus_has_invalid_primary()
823 uint32_t start_layer, uint32_t num_layers, in crocus_resource_prepare_access()
881 uint32_t start_layer, uint32_t num_layers, in crocus_resource_finish_write()
[all …]
/third_party/mindspore/tests/st/networks/
Dtest_gpu_lstm.py31 def InitialLstmWeight(input_size, hidden_size, num_layers, bidirectional, has_bias=False): argument
62 def __init__(self, vocab_size, embed_size, num_hiddens, num_layers, argument
/third_party/mindspore/tests/st/ops/ascend/
Dtest_gru_op.py40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
Dtest_rnn_op.py40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
Dtest_lstm_op.py41 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
51 def __init__(self, num_layers, has_bias, input_s, num_directions, hidden_s, bidirectional): argument
/third_party/mindspore/mindspore/core/ops/grad/
Dlstm_grad.cc44 void LSTMGrad::set_num_layers(const int64_t num_layers) { in set_num_layers()
79 void LSTMGrad::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, … in Init()
/third_party/mindspore/mindspore/core/ops/
Dlstm.cc54 int64_t num_layers = GetValue<int64_t>(primitive->GetAttr(kNumLayers)); in LstmInfer() local
116 void LSTM::set_num_layers(const int64_t num_layers) { in set_num_layers()
151 void LSTM::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, cons… in Init()
/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_for_loop_reshape.py92 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
111 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
Dtest_auto_parallel_for_loop.py85 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
104 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
Dtest_auto_parallel_for_loop_simplify.py62 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
/third_party/mesa3d/src/intel/blorp/
Dblorp_clear.c426 uint32_t level, uint32_t start_layer, uint32_t num_layers, in blorp_fast_clear()
489 uint32_t level, uint32_t start_layer, uint32_t num_layers, in blorp_clear()
684 uint32_t num_layers, in blorp_clear_stencil_as_rgba()
789 uint32_t num_layers, in blorp_clear_depth_stencil()
1008 uint32_t start_layer, uint32_t num_layers, in blorp_hiz_clear_depth_stencil()
1112 uint32_t start_layer, uint32_t num_layers, in blorp_clear_attachments()
1176 uint32_t start_layer, uint32_t num_layers, in blorp_ccs_resolve()
1355 uint32_t start_layer, uint32_t num_layers) in blorp_mcs_partial_resolve()
/third_party/mindspore/tests/ut/python/nn/
Dtest_lstm.py27 def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional): argument
/third_party/mesa3d/src/gallium/winsys/radeon/drm/
Dradeon_drm_surface.c271 unsigned num_layers; in si_compute_cmask() local
284 struct radeon_surf *surf, unsigned num_layers) in si_compute_htile()
/third_party/mesa3d/src/mesa/state_tracker/
Dst_draw_feedback.c328 unsigned num_layers = res->depth0; in st_feedback_draw_vbo() local
407 unsigned width, height, num_layers, row_stride, img_stride; in st_feedback_draw_vbo() local
/third_party/mesa3d/src/mesa/drivers/dri/i965/
Dbrw_clear.c174 const uint32_t num_layers = depth_att->Layered ? depth_irb->layer_count : 1; in brw_fast_clear_depth() local
/third_party/freetype/src/sfnt/
Dttcolr.c51 FT_UShort num_layers; member
60 FT_UShort num_layers; member
/third_party/flutter/skia/third_party/externals/freetype/src/sfnt/
Dttcolr.c52 FT_UShort num_layers; member
61 FT_UShort num_layers; member
/third_party/mesa3d/src/gallium/drivers/radeonsi/
Dsi_clear.c308 unsigned num_layers = util_num_layers(&tex->buffer.b.b, level); in vi_dcc_get_clear_info() local
341 unsigned num_layers = util_num_layers(&tex->buffer.b.b, level); in vi_dcc_get_clear_info() local
561 unsigned num_layers = util_num_layers(&tex->buffer.b.b, level); in si_fast_clear() local
/third_party/mesa3d/src/gallium/auxiliary/util/
Du_framebuffer.c184 unsigned i, num_layers = 0; in util_framebuffer_get_num_layers() local
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/
Drpn.py152 … def _make_rpn_layer(self, num_layers, in_channels, feat_channels, num_anchors, cls_out_channels): argument
/third_party/skia/third_party/externals/freetype/src/sfnt/
Dttcolr.c68 FT_UShort num_layers; member
86 FT_UShort num_layers; member
455 FT_Byte num_layers; in read_paint() local
/third_party/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_sampler.c298 unsigned num_layers = tex->depth0; in prepare_shader_sampling() local
474 unsigned num_layers = img->depth0; in prepare_shader_images() local

123