Searched refs:second_signature (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 152 const OpDef& second_signature, in GetUniqueSignature() argument
156 GetUniqueNames(first_signature.input_arg(), second_signature.input_arg()); in GetUniqueSignature()
158 signature.set_name(second_signature.name()); in GetUniqueSignature()
160 for (const auto& input_arg : second_signature.input_arg()) { in GetUniqueSignature()
169 first_signature.output_arg(), second_signature.output_arg()); in GetUniqueSignature()
171 for (const auto& output_arg : second_signature.output_arg()) { in GetUniqueSignature()
288 const OpDef& second_signature) { in CheckIfCanCompose() argument
289 CHECK(CanCompose(first_signature, second_signature)) in CheckIfCanCompose()
290 << "The number of input arguments of function " << second_signature.name() in CheckIfCanCompose()
306 bool CanCompose(const OpDef& first_signature, const OpDef& second_signature) { in CanCompose() argument
[all …]
|
| D | fusion_utils.h | 67 bool CanCompose(const OpDef& first_signature, const OpDef& second_signature);
70 const OpDef& second_signature, OpDef* fused_signature);
85 const OpDef& second_signature, OpDef* fused_signature);
97 const OpDef& second_signature);
100 void SameSignature(const OpDef& first_signature, const OpDef& second_signature,
|