Searched refs:weights_name (Results 1 – 8 of 8) sorted by relevance
30 const std::string& weights_name = op.inputs[1]; in GetOutputDepthFromWeights() local31 const auto& weights_shape = model.GetArray(weights_name).shape(); in GetOutputDepthFromWeights()60 const std::string& weights_name = op->inputs[1]; in ProcessLinearOperator() local61 if (!model->GetArray(weights_name).has_shape()) { in ProcessLinearOperator()
55 const auto& weights_name = conv_op->inputs[1]; in Run() local56 if (CountOpsWithInput(*model, weights_name) > 1) { in Run()64 auto& weights_array = model->GetArray(weights_name); in Run()
40 const std::string& weights_name = fc_op->inputs[1]; in Run() local41 Array& weights_array = model->GetArray(weights_name); in Run()109 if (CountOpsWithInput(*model, weights_name) != 1) { in Run()
116 const auto& weights_name = preceding_op->inputs[1]; in FuseMulOrDivParamsIntoPrecedingAffine() local119 auto& weights = model->GetArray(weights_name); in FuseMulOrDivParamsIntoPrecedingAffine()120 DropMinMax(model, weights_name); in FuseMulOrDivParamsIntoPrecedingAffine()303 const auto& weights_name = preceding_op->inputs[1]; in Run() local306 const auto& weights = model->GetArray(weights_name); in Run()322 CountOpsWithInput(*model, weights_name); in Run()
123 const auto& weights_name = following_op->inputs[1]; in FuseMulOrDivParamsIntoFollowingAffine() local125 auto& weights = model->GetArray(weights_name); in FuseMulOrDivParamsIntoFollowingAffine()126 DropMinMax(model, weights_name); in FuseMulOrDivParamsIntoFollowingAffine()
841 const auto& weights_name = node.input(1); in ConvertConvOperator() local843 AvailableArrayName(*model, weights_name + "_reordered"); in ConvertConvOperator()855 reorder->inputs = {weights_name}; in ConvertConvOperator()924 const auto& weights_name = node.input(1); in ConvertDepthwiseConvOperator() local925 const auto& reordered_weights_name = weights_name + "_reordered"; in ConvertDepthwiseConvOperator()937 reorder->inputs = {weights_name}; in ConvertDepthwiseConvOperator()2009 const std::string& weights_name = node.input(TransposeConvOperator::WEIGHTS); in ConvertTransposeConvOperator() local2010 const std::string& transposed_weights_name = weights_name + "_transposed"; in ConvertTransposeConvOperator()2025 transpose->inputs = {weights_name, perm_array}; in ConvertTransposeConvOperator()
2425 const std::string& weights_name = fc_op.inputs[1]; in UndoWeightsShuffling() local2426 QCHECK_EQ(CountOpsWithInput(*model, weights_name), 1); in UndoWeightsShuffling()2427 auto& weights_array = model->GetArray(weights_name); in UndoWeightsShuffling()
1510 const std::string weights_name = WalkUpToConstantArray( in ConvertLstmCellOperator() local1512 const auto& weights_array = model.GetArray(weights_name); in ConvertLstmCellOperator()