Searched refs:following_op (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dfuse_binary_into_following_affine.cc31 void FuseAddOrSubParamsIntoFollowingAffine(Model* model, Operator* following_op, argument
42 if (following_op->inputs.size() < 3) {
45 const auto& weights = model->GetArray(following_op->inputs[1]);
46 auto& bias = model->GetArray(following_op->inputs[2]);
77 if (following_op->type == OperatorType::kConv ||
78 following_op->type == OperatorType::kFullyConnected) {
94 } else if (following_op->type == OperatorType::kDepthwiseConv) {
113 void FuseMulOrDivParamsIntoFollowingAffine(Model* model, Operator* following_op, argument
123 const auto& weights_name = following_op->inputs[1];
124 const auto& bias_name = following_op
218 Operator* following_op = GetOpWithInput(*model, binary_op->outputs[0]); local
[all...]

Completed in 325 milliseconds