Home
last modified time | relevance | path

Searched refs:weight_variable_1 (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_batchnorm_batch_parallel.py76 def weight_variable_1(shape): function
84 var = weight_variable_1(shape)
86 gamma = weight_variable_1(shape)
Dtest_operator_model_parallel.py142 def weight_variable_1(shape): function
150 var = weight_variable_1(shape)
152 gamma = weight_variable_1(shape)
162 var = weight_variable_1(shape)
/third_party/mindspore/tests/st/networks/models/
Dresnetv1_5.py32 def weight_variable_1(shape): function
64 var = weight_variable_1(shape)
66 gamma = weight_variable_1(shape)
75 var = weight_variable_1(shape)
/third_party/mindspore/tests/st/export_and_load/
Dtest_resnet_air.py50 def weight_variable_1(shape): function
82 var = weight_variable_1(shape)
84 gamma = weight_variable_1(shape)
93 var = weight_variable_1(shape)
/third_party/mindspore/tests/st/networks/
Dtest_gpu_resnet.py56 def weight_variable_1(shape): function
88 var = weight_variable_1(shape)
90 gamma = weight_variable_1(shape)
99 var = weight_variable_1(shape)
/third_party/mindspore/tests/st/tbe_networks/
Dresnet.py27 def weight_variable_1(shape): function
56 var = weight_variable_1(shape)
66 var = weight_variable_1(shape)
/third_party/mindspore/tests/st/mem_reuse/
Dresnet.py37 def weight_variable_1(shape): function
69 var = weight_variable_1(shape)
80 var = weight_variable_1(shape)
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_resnet50_gpu.py50 def weight_variable_1(shape): function
79 var = weight_variable_1(shape)
89 var = weight_variable_1(shape)
Dtest_pynative_resnet50_ascend.py88 def weight_variable_1(shape): function
117 var = weight_variable_1(shape)
127 var = weight_variable_1(shape)