Searched refs:GetOpWithInput (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_dilated_conv.cc | 71 auto* post_conv_op = GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
86 ? GetOpWithInput(*model, post_conv_op->outputs[0]) in ResolveDilatedConv()
87 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
98 ? GetOpWithInput(*model, pad_op->outputs[0]) in ResolveDilatedConv()
100 ? GetOpWithInput(*model, post_conv_op->outputs[0]) in ResolveDilatedConv()
101 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
106 auto final_op = GetOpWithInput(*model, next_op->outputs[0]); in ResolveDilatedConv()
197 auto* post_stb_op = GetOpWithInput(*model, stb_op->outputs[0]); in Run()
211 auto* conv_base_op = GetOpWithInput(*model, input_of_conv_op); in Run()
|
| D | identify_hardswish.cc | 49 const auto* mul_op = GetOpWithInput(*model, add_with_relu6_op->outputs[0]); in Run()
53 mul_op = GetOpWithInput(*model, mul_op->outputs[0]); in Run()
60 const auto* output_op = GetOpWithInput(*model, mul_op->outputs[0]); in Run()
63 output_op = GetOpWithInput(*model, output_op->outputs[0]); in Run()
|
| D | resolve_squeeze_attributes.cc | 41 const auto* next_op = GetOpWithInput(*model, squeeze_op->outputs[0]); in Run()
|
| D | remove_trivial_reshape.cc | 60 const auto* next_op = GetOpWithInput(model, op.outputs[0]); in IsReshapeTrivial()
|
| D | identify_relu1.cc | 42 const auto* op_1 = GetOpWithInput(*model, op_0->outputs[0]); in Run()
|
| D | remove_successive_transpose.cc | 72 Operator* next = GetOpWithInput(*model, t_op->outputs[0]); in Run()
|
| D | resolve_tensorflow_switch.cc | 119 if (!GetOpWithInput(*model, switch_op->outputs[i])) { in Run()
|
| D | fuse_binary_into_following_affine.cc | 227 Operator* following_op = GetOpWithInput(*model, binary_op->outputs[0]); in Run()
|
| D | group_bidirectional_sequence_ops.cc | 379 Operator* unpack_following_op = GetOpWithInput(*model, output_array_name); in RewireFinalUnpackOutputs()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 86 Operator* GetOpWithInput(const Model& model, const std::string& array_name);
|
| D | tooling_util.cc | 118 if (GetOpWithInput(model, name)) { in IsArrayConsumed()
176 const Operator* op_having_this_as_input = GetOpWithInput(*model, array_name); in DeleteArrayIfUnusedOutsideOfOp()
275 Operator* GetOpWithInput(const Model& model, const std::string& array_name) { in GetOpWithInput() function
948 QCHECK(GetOpWithInput(model, rnn_state.state_array())) in CheckNonExistentIOArrays()
|