Searched refs:GetOpWithInput (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_dilated_conv.cc | 68 auto* post_conv_op = GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
83 ? GetOpWithInput(*model, post_conv_op->outputs[0]) in ResolveDilatedConv()
84 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
95 ? GetOpWithInput(*model, pad_op->outputs[0]) in ResolveDilatedConv()
97 ? GetOpWithInput(*model, post_conv_op->outputs[0]) in ResolveDilatedConv()
98 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
103 auto final_op = GetOpWithInput(*model, next_op->outputs[0]); in ResolveDilatedConv()
199 auto* post_stb_op = GetOpWithInput(*model, stb_op->outputs[0]); in Run()
213 auto* conv_base_op = GetOpWithInput(*model, input_of_conv_op); 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 | resolve_tensorflow_switch.cc | 110 if (!GetOpWithInput(*model, switch_op->outputs[i])) { in Run()
|
| D | identify_relu1.cc | 71 const auto* op_1 = GetOpWithInput(*model, op_0->outputs[0]); 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 | 377 Operator* unpack_following_op = GetOpWithInput(*model, output_array_name); in RewireFinalUnpackOutputs()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 90 Operator* GetOpWithInput(const Model& model, const string& array_name);
|
| D | tooling_util.cc | 118 if (GetOpWithInput(model, name)) { in IsArrayConsumed()
259 Operator* GetOpWithInput(const Model& model, const string& array_name) { in GetOpWithInput() function
919 QCHECK(GetOpWithInput(model, rnn_state.state_array())) in CheckNonExistentIOArrays()
|