Home
last modified time | relevance | path

Searched defs:input_channels (Results 1 – 25 of 66) sorted by relevance

123

/third_party/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_ml_nn_v7.c69 unsigned input_channels = operation->input_channels; in etna_ml_calculate_tiling_v7() local
190 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in write_core_6() local
270 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in write_core_interleaved() local
445 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in calculate_weight_bo_size() local
470 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in calculate_zrl_bits() local
522 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in etna_ml_create_coeffs_v7() local
Detnaviv_ml_nn_v8.c112 unsigned input_channels = operation->input_channels; in etna_ml_calculate_tiling_v8() local
184 unsigned input_channels; in calculate_bias_correction() local
624 unsigned input_channels = operation->input_channels; in encode_superblock() local
695 unsigned input_channels = operation->input_channels; in create_bo() local
716 unsigned input_channels = operation->input_channels; in calculate_symbol_map() local
790 unsigned input_channels = operation->input_channels; in fill_biases() local
Detnaviv_ml_nn.c281 unsigned input_channels; in transpose() local
407 unsigned input_channels = poperation->input_tensors[0]->dims[3]; in calc_pooling_first_pixel() local
706 etna_ml_calc_addition_sizes(unsigned *input_width, unsigned *input_height, unsigned *input_channels, in etna_ml_calc_addition_sizes()
758 unsigned input_channels = operation->input_channels; in create_nn_config() local
/third_party/mindspore/mindspore-src/source/tests/ut/python/exec/
Dtest_dense.py27 input_channels, argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_tbe_ops/
Dtest_apply_adam.py27 … def __init__(self, batch_num, input_channels, output_channels, epoch, lr, weight_decay, epsilon): argument
/third_party/mindspore/mindspore-src/source/tests/perf_test/
Dlenet.py28 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/st/export_and_load/
Dtest_dynamic_obfuscation_st.py39 def fc_with_initialize(input_channels, out_channels): argument
Dtest_export_mindir_choice_func.py36 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/export_models/models/
Dmini_alexnet.py26 def fc_with_initialize(input_channels, out_channels, has_bias=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/
Dtest_construct.py19 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/st/rewrite/
Dtest_rewrite_obfuscate_weight.py36 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/nn/
Dtest_dense.py94 input_channels, argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/rewrite/test_obfuscate/
Dtest_obfuscate_weight.py31 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/st/profiler/
Drun_net_with_profiler.py39 def fc_with_initialize(input_channels, out_channels): argument
Drun_net.py38 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mesa3d/src/gallium/targets/teflon/
Dtest_teflon.cpp34 std::vector<int> input_channels{1, 32, 120, 128, 256}; variable
142 test_conv(int input_size, int weight_size, int input_channels, int output_channels, in test_conv()
185 test_add(int input_size, int weight_size, int input_channels, int output_channels, in test_add()
608 int input_channels = atoi(argv[n++]); in main() local
Dtest_executor.cpp54 int input_channels, in patch_conv2d()
166 int input_channels, in conv2d_generate_model()
214 int input_channels, in add_generate_model()
/third_party/mindspore/mindspore-src/source/tests/ut/python/mindir/
Dtest_dynamic_obfuscation.py39 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/
Ddense.py50 def check_last_dimension(input_dim, input_channels, input_name, input_channels_name, prim_name=None… argument
/third_party/mindspore/mindspore-src/source/tests/st/train/
Dtest_on_request_exit_callback.py41 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/utils/
Dtest_get_func_graph.py35 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/st/compile_cache/
Drun_lenet_ps.py44 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/st/armour/
Dtest_pynative_mindarmour.py42 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/st/probability/bnn_layers/
Dtest_bnn_layer.py38 def fc_with_initialize(input_channels, out_channels): argument
/third_party/mindspore/mindspore-src/source/tests/st/place/general_split/src/
Dlenet.py70 def fc_with_initialize(input_channels, out_channels): argument

123