Searched refs:fc_op (Results 1 – 6 of 6) sorted by relevance
34 FullyConnectedOperator* fc_op = static_cast<FullyConnectedOperator*>(op); in Run() local36 if (fc_op->weights_format != FullyConnectedWeightsFormat::kDefault) { in Run()39 const Array& input_array = model->GetArray(fc_op->inputs[0]); in Run()40 const string& weights_name = fc_op->inputs[1]; in Run()42 const Array& output_array = model->GetArray(fc_op->outputs[0]); in Run()140 fc_op->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8; in Run()145 CHECK_EQ(fc_op->outputs.size(), 1); in Run()147 AvailableArrayName(*model, fc_op->inputs[0] + "_shuffled"); in Run()148 fc_op->outputs.push_back(shuffled_input_workspace_array_name); in Run()
131 auto* fc_op = new FullyConnectedOperator; in Run() local132 fc_op->outputs = matmul_op->outputs; in Run()135 model->operators.emplace(matmul_it, fc_op) + 1; in Run()169 LogName(*matmul_op), LogName(*fc_op)); in Run()200 fc_op->inputs = {input_lhs, input_rhs}; in Run()
125 const auto& fc_op = static_cast<const toco::FullyConnectedOperator&>(op); in Run() local126 CHECK(fc_op.weights_format == FullyConnectedWeightsFormat::kDefault) in Run()
87 auto* fc_op = new FullyConnectedOperator; in TEST_F() local88 fc_op->inputs = {"MulOutput", "FCWeight", "FCBias"}; in TEST_F()89 fc_op->outputs = {"Output"}; in TEST_F()90 model_->operators.push_back(std::unique_ptr<Operator>(fc_op)); in TEST_F()129 auto* fc_op = new FullyConnectedOperator; in TEST_F() local130 fc_op->inputs = {"MulOutput", "FCWeight", "FCBias"}; in TEST_F()131 fc_op->outputs = {"Output"}; in TEST_F()132 model_->operators.push_back(std::unique_ptr<Operator>(fc_op)); in TEST_F()
2359 const auto& fc_op = static_cast<toco::FullyConnectedOperator&>(*op); in UndoWeightsShuffling() local2360 if (fc_op.weights_format == FullyConnectedWeightsFormat::kDefault) { in UndoWeightsShuffling()2363 const string& weights_name = fc_op.inputs[1]; in UndoWeightsShuffling()
489 const auto& fc_op = in GetVersion() local512 if (fc_op.weights_format == in GetVersion()