Searched refs:fc_op (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 212 auto fc_op = in matchAndRewrite() local
214 if (!fc_op) return matchFailure(); in matchAndRewrite()
231 Value filter = fc_op.filter(); in matchAndRewrite()
232 Value bias = fc_op.bias(); in matchAndRewrite()
235 if (fc_op.fused_activation_function() != "NONE") return matchFailure(); in matchAndRewrite()
241 Location loc = fc_op.getLoc(); in matchAndRewrite()
281 /*input=*/fc_op.input(), in matchAndRewrite()
286 /*weights_format=*/rewriter.getStringAttr(fc_op.weights_format()), in matchAndRewrite()
287 /*keep_num_dims=*/rewriter.getBoolAttr(fc_op.keep_num_dims())); in matchAndRewrite()
332 auto fc_op = in matchAndRewrite() local
[all …]
|
| D | legalize_tf.cc | 174 auto fc_op = rewriter.create<FullyConnectedOp>( in matchAndRewrite() local
178 rewriter.replaceOp(op, {fc_op.getResult(0)}); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | shuffle_fc_weights.cc | 34 FullyConnectedOperator* fc_op = static_cast<FullyConnectedOperator*>(op); in Run() local
36 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()
|
| D | resolve_tensorflow_matmul.cc | 167 auto* fc_op = new FullyConnectedOperator; in Run() local
168 fc_op->inputs = {input_lhs, input_rhs}; in Run()
169 fc_op->outputs = matmul_op->outputs; in Run()
172 model->operators.emplace(matmul_it, fc_op) + 1; in Run()
206 LogName(*matmul_op), LogName(*fc_op)); in Run()
213 fc_op->inputs = {input_lhs, input_rhs}; in Run()
|
| D | ensure_uint8_weights_safe_for_fast_int8_kernels.cc | 125 const auto& fc_op = static_cast<const toco::FullyConnectedOperator&>(op); in Run() local
126 CHECK(fc_op.weights_format == FullyConnectedWeightsFormat::kDefault) in Run()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | fuse_binary_into_following_affine_test.cc | 87 auto* fc_op = new FullyConnectedOperator; in TEST_F() local
88 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() local
130 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()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 428 const auto& fc_op = in GetVersion() local
432 op_sig.options.fully_connected.keep_num_dims = fc_op.keep_num_dims; in GetVersion()
434 GetWeightFormat(fc_op.weights_format); in GetVersion()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2401 const auto& fc_op = static_cast<toco::FullyConnectedOperator&>(*op); in UndoWeightsShuffling() local
2402 if (fc_op.weights_format == FullyConnectedWeightsFormat::kDefault) { in UndoWeightsShuffling()
2405 const string& weights_name = fc_op.inputs[1]; in UndoWeightsShuffling()
|