HomeSort by relevance Sort by last modified time
    Searched refs:fc_op (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
shuffle_fc_weights.cc 34 FullyConnectedOperator* fc_op = static_cast<FullyConnectedOperator*>(op); local
36 if (fc_op->weights_format != FullyConnectedWeightsFormat::kDefault) {
39 const Array& input_array = model->GetArray(fc_op->inputs[0]);
40 const string& weights_name = fc_op->inputs[1];
42 const Array& output_array = model->GetArray(fc_op->outputs[0]);
140 fc_op->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8;
145 CHECK_EQ(fc_op->outputs.size(), 1);
147 AvailableArrayName(*model, fc_op->inputs[0] + "_shuffled");
148 fc_op->outputs.push_back(shuffled_input_workspace_array_name);
ensure_uint8_weights_safe_for_fast_int8_kernels.cc 125 const auto& fc_op = static_cast<const toco::FullyConnectedOperator&>(op); local
126 CHECK(fc_op.weights_format == FullyConnectedWeightsFormat::kDefault)
resolve_tensorflow_matmul.cc 131 auto* fc_op = new FullyConnectedOperator; local
132 fc_op->outputs = matmul_op->outputs;
135 model->operators.emplace(matmul_it, fc_op) + 1;
169 LogName(*matmul_op), LogName(*fc_op));
200 fc_op->inputs = {input_lhs, input_rhs};
  /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
fuse_binary_into_following_affine_test.cc 87 auto* fc_op = new FullyConnectedOperator; local
88 fc_op->inputs = {"MulOutput", "FCWeight", "FCBias"};
89 fc_op->outputs = {"Output"};
90 model_->operators.push_back(std::unique_ptr<Operator>(fc_op));
129 auto* fc_op = new FullyConnectedOperator; local
130 fc_op->inputs = {"MulOutput", "FCWeight", "FCBias"};
131 fc_op->outputs = {"Output"};
132 model_->operators.push_back(std::unique_ptr<Operator>(fc_op));
  /external/tensorflow/tensorflow/lite/toco/
tooling_util.cc 2359 const auto& fc_op = static_cast<toco::FullyConnectedOperator&>(*op); local
    [all...]
  /external/tensorflow/tensorflow/lite/toco/tflite/
operator.cc 489 const auto& fc_op = variable
512 if (fc_op.weights_format ==
    [all...]

Completed in 1851 milliseconds