• Home
  • Raw
  • Download

Lines Matching refs:set_name

85               .set_name("conv1/convolution")  in do_setup()
92 .set_name("conv1/BatchNorm") in do_setup()
93 …ionLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name("conv1/Relu") in do_setup()
94 …_layout, PadStrideInfo(2, 2, 0, 1, 0, 1, DimensionRoundingType::FLOOR))).set_name("pool1/MaxPool"); in do_setup()
101 … graph << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, operation_layout)).set_name("pool5") in do_setup()
107 .set_name("logits/convolution") in do_setup()
108 << FlattenLayer().set_name("predictions/Reshape") in do_setup()
109 << SoftmaxLayer().set_name("predictions/Softmax") in do_setup()
163 .set_name(unit_name + "conv1/convolution") in add_residual_block()
170 .set_name(unit_name + "conv1/BatchNorm") in add_residual_block()
171 …yer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(unit_name + "conv… in add_residual_block()
178 .set_name(unit_name + "conv2/convolution") in add_residual_block()
185 .set_name(unit_name + "conv2/BatchNorm") in add_residual_block()
186 …yer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(unit_name + "conv… in add_residual_block()
193 .set_name(unit_name + "conv3/convolution") in add_residual_block()
200 .set_name(unit_name + "conv2/BatchNorm"); in add_residual_block()
210 .set_name(unit_name + "shortcut/convolution") in add_residual_block()
217 .set_name(unit_name + "shortcut/BatchNorm"); in add_residual_block()
219 … EltwiseLayer(std::move(left), std::move(right), EltwiseOperation::Add).set_name(unit_name + "add"… in add_residual_block()
224 ….data_layout, PadStrideInfo(middle_stride, middle_stride, 0, 0), true)).set_name(unit_name + "shor… in add_residual_block()
226 … EltwiseLayer(std::move(left), std::move(right), EltwiseOperation::Add).set_name(unit_name + "add"… in add_residual_block()
231 … EltwiseLayer(std::move(left), std::move(right), EltwiseOperation::Add).set_name(unit_name + "add"… in add_residual_block()
234 …yer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)).set_name(unit_name + "Relu… in add_residual_block()