Home
last modified time | relevance | path

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

1234

/third_party/mesa3d/src/gallium/drivers/iris/
Diris_resolve.c138 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 …]
/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()
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 …]
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dgru_v2_grad.cc38 void GRUV2Grad::set_num_layers(const int64_t num_layers) { in set_num_layers()
68 void GRUV2Grad::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers,… in Init()
114 int64_t num_layers = GetValue<int64_t>(primitive->GetAttr(kNumLayers)); in InferShape() local
139 …int64_t GetWeightSize(const PrimitivePtr &primitive, int64_t num_layers, int64_t num_directions) c… in GetWeightSize()
Dlstm_grad.cc67 void LSTMGrad::set_num_layers(const int64_t num_layers) { in set_num_layers()
106 void LSTMGrad::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, … in Init()
154 int64_t num_layers = GetValue<int64_t>(primitive->GetAttr(kNumLayers)); in InferShape() local
199 …int64_t GetWeightSize(const PrimitivePtr &primitive, int64_t num_layers, int64_t num_directions) c… in GetWeightSize()
Dlstm_grad_weight.cc52 void LSTMGradWeight::set_num_layers(const int64_t num_layers) { in set_num_layers()
98 …LSTMGradWeight::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, in Init()
124 int64_t num_layers = GetValue<int64_t>(primitive->GetAttr(kNumLayers)); in LstmGradWeightInferShape() local
Dlstm_grad_data.cc55 void LSTMGradData::set_num_layers(const int64_t num_layers) { in set_num_layers()
101 …d LSTMGradData::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, in Init()
148 int64_t num_layers = GetValue<int64_t>(primitive->GetAttr(kNumLayers)); in LstmGradDataInferShape() local
/third_party/mindspore/mindspore-src/source/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/mindspore-src/source/tests/st/ops/cpu/
Dtest_lstm_op.py41 …def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout, p… argument
58 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
68 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
/third_party/mindspore/mindspore-src/source/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.py40 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_lstm_op.py40 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
Dtest_gruv2_op.py27 def __init__(self, input_size, hidden_size, num_layers, has_bias, bidirectional): argument
37 def __init__(self, input_size, hidden_size, num_layers, has_bias, weights, is_train): argument
Dtest_lstmv2_op.py27 def __init__(self, input_size, hidden_size, num_layers, has_bias, bidirectional): argument
37 def __init__(self, input_size, hidden_size, num_layers, has_bias, weights, is_train): argument
/third_party/mesa3d/src/intel/blorp/
Dblorp_clear.c443 uint32_t level, uint32_t start_layer, uint32_t num_layers, in blorp_fast_clear()
573 uint32_t level, uint32_t start_layer, uint32_t num_layers, in blorp_clear()
786 uint32_t num_layers, in blorp_clear_stencil_as_rgba()
891 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()
1384 uint32_t start_layer, uint32_t num_layers) in blorp_mcs_partial_resolve()
1454 uint32_t start_layer, uint32_t num_layers) in blorp_mcs_ambiguate()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/
Dpytorch_lstm_parser.cc39 …auto num_layers = PytorchNodeParser::GetValueFromConstNode<int64_t>(torch_node->input(FIFTH_INPUT)… in Parse() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dlstm.cc69 int64_t num_layers = GetValue<int64_t>(primitive->GetAttr(kNumLayers)); in LSTMInferShape() local
155 void LSTM::set_num_layers(const int64_t num_layers) { in set_num_layers()
200 void LSTM::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, cons… in Init()
/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_auto_parallel_for_loop.py94 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
113 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
Dtest_auto_parallel_for_loop_reshape.py98 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
117 def __init__(self, mul_weight, num_layers, strategy1=None, strategy2=None): argument
/third_party/mindspore/mindspore-src/source/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/drivers/svga/include/
Dvmw_surf_defs.h111 uint32 num_layers) in vmw_surf_get_serialized_size()
135 uint32 num_mip_levels, uint32 num_layers, in vmw_surf_get_serialized_size_extended()
/third_party/mesa3d/src/gallium/winsys/radeon/drm/
Dradeon_drm_surface.c251 unsigned num_layers; in si_compute_cmask() local
264 struct radeon_surf *surf, unsigned num_layers) in si_compute_htile()
/third_party/mindspore/mindspore-src/source/tests/st/networks/models/fasterrcnn/src/rpn_head/
Drpn_head.py33 def __init__(self, num_layers=1, num_anchors=3, in_channel=1024, out_channel=1024): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/ops/
Dtest_dyn_shape_primitive.py108 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
118 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
/third_party/mesa3d/src/mesa/state_tracker/
Dst_draw_feedback.c276 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

1234