Searched refs:scale_values (Results 1 – 2 of 2) sorted by relevance
42 std::vector<float>* scale_values, in GetScaleAndOffsetValues() argument83 scale_values->resize(num_cols); in GetScaleAndOffsetValues()89 (*scale_values)[i] = in GetScaleAndOffsetValues()95 (*scale_values)[i] = in GetScaleAndOffsetValues()101 (-mean.flat<float>()(i) * (*scale_values)[i]) + beta.flat<float>()(i); in GetScaleAndOffsetValues()106 Status FuseScaleOffsetToConvWeights(const std::vector<float>& scale_values, in FuseScaleOffsetToConvWeights() argument126 CHECK_EQ(weights_cols, scale_values.size()); in FuseScaleOffsetToConvWeights()134 weights_vector(row) * scale_values[row % weights_cols]; in FuseScaleOffsetToConvWeights()179 std::vector<float> scale_values; in FuseBatchNormWithConv() local182 GetScaleAndOffsetValues(match, &scale_values, &offset_values)); in FuseBatchNormWithConv()[all …]
88 CenterSizeEncoding scale_values; member114 op_data->scale_values.y = m["y_scale"].AsFloat(); in Init()115 op_data->scale_values.x = m["x_scale"].AsFloat(); in Init()116 op_data->scale_values.h = m["h_scale"].AsFloat(); in Init()117 op_data->scale_values.w = m["w_scale"].AsFloat(); in Init()271 CenterSizeEncoding scale_values = op_data->scale_values; in DecodeCenterSizeBoxes() local299 float ycenter = box_centersize.y / scale_values.y * anchor.h + anchor.y; in DecodeCenterSizeBoxes()300 float xcenter = box_centersize.x / scale_values.x * anchor.w + anchor.x; in DecodeCenterSizeBoxes()302 0.5f * static_cast<float>(std::exp(box_centersize.h / scale_values.h)) * in DecodeCenterSizeBoxes()305 0.5f * static_cast<float>(std::exp(box_centersize.w / scale_values.w)) * in DecodeCenterSizeBoxes()