• Home
  • Raw
  • Download

Lines Matching refs:data_path

62         std::string data_path  = common_params.data_path;  in do_setup()  local
64 if(!data_path.empty()) in do_setup()
66 data_path += model_path; in do_setup()
85 get_weights_accessor(data_path, "conv1_weights.npy", weights_layout), in do_setup()
86 get_weights_accessor(data_path, "conv1_biases.npy", weights_layout), in do_setup()
91 add_residual_block(data_path, "block1", weights_layout, 64, 3, 2); in do_setup()
92 add_residual_block(data_path, "block2", weights_layout, 128, 4, 2); in do_setup()
93 add_residual_block(data_path, "block3", weights_layout, 256, 6, 2); in do_setup()
94 add_residual_block(data_path, "block4", weights_layout, 512, 3, 1); in do_setup()
97 get_weights_accessor(data_path, "postnorm_moving_mean.npy"), in do_setup()
98 get_weights_accessor(data_path, "postnorm_moving_variance.npy"), in do_setup()
99 get_weights_accessor(data_path, "postnorm_gamma.npy"), in do_setup()
100 get_weights_accessor(data_path, "postnorm_beta.npy"), in do_setup()
107 get_weights_accessor(data_path, "logits_weights.npy", weights_layout), in do_setup()
108 get_weights_accessor(data_path, "logits_biases.npy"), in do_setup()
140 …void add_residual_block(const std::string &data_path, const std::string &name, DataLayout weights_… in add_residual_block() argument
164 get_weights_accessor(data_path, unit_path + "preact_moving_mean.npy"), in add_residual_block()
165 get_weights_accessor(data_path, unit_path + "preact_moving_variance.npy"), in add_residual_block()
166 get_weights_accessor(data_path, unit_path + "preact_gamma.npy"), in add_residual_block()
167 get_weights_accessor(data_path, unit_path + "preact_beta.npy"), in add_residual_block()
186 … get_weights_accessor(data_path, unit_path + "shortcut_weights.npy", weights_layout), in add_residual_block()
187 … get_weights_accessor(data_path, unit_path + "shortcut_biases.npy", weights_layout), in add_residual_block()
196 … get_weights_accessor(data_path, unit_path + "conv1_weights.npy", weights_layout), in add_residual_block()
201 … get_weights_accessor(data_path, unit_path + "conv1_BatchNorm_moving_mean.npy"), in add_residual_block()
202 … get_weights_accessor(data_path, unit_path + "conv1_BatchNorm_moving_variance.npy"), in add_residual_block()
203 get_weights_accessor(data_path, unit_path + "conv1_BatchNorm_gamma.npy"), in add_residual_block()
204 get_weights_accessor(data_path, unit_path + "conv1_BatchNorm_beta.npy"), in add_residual_block()
210 … get_weights_accessor(data_path, unit_path + "conv2_weights.npy", weights_layout), in add_residual_block()
215 … get_weights_accessor(data_path, unit_path + "conv2_BatchNorm_moving_mean.npy"), in add_residual_block()
216 … get_weights_accessor(data_path, unit_path + "conv2_BatchNorm_moving_variance.npy"), in add_residual_block()
217 get_weights_accessor(data_path, unit_path + "conv2_BatchNorm_gamma.npy"), in add_residual_block()
218 get_weights_accessor(data_path, unit_path + "conv2_BatchNorm_beta.npy"), in add_residual_block()
224 … get_weights_accessor(data_path, unit_path + "conv3_weights.npy", weights_layout), in add_residual_block()
225 … get_weights_accessor(data_path, unit_path + "conv3_biases.npy", weights_layout), in add_residual_block()