| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| identify_prelu.cc | 57 const auto* relu_input_op = GetOpWithOutput(*model, add_op->inputs[0]);
67 const auto* mul_op = GetOpWithOutput(*model, add_op->inputs[1]);
76 const auto* relu_neg_input_op = GetOpWithOutput(*model, mul_op->inputs[1]);
92 GetOpWithOutput(*model, relu_neg_input_op->inputs[0]);
|
| fuse_broadcast_into_following_binary.cc | 74 GetOpWithOutput(*model, binary_op->inputs[0]),
75 GetOpWithOutput(*model, binary_op->inputs[1]),
|
| identify_l2_normalization.cc | 58 GetOpWithOutput(*model, div_or_mul_op->inputs[0]),
59 GetOpWithOutput(*model, div_or_mul_op->inputs[1]),
69 GetOpWithOutput(*model, sqrt_or_rsqrt_op->inputs[0]);
102 op_producing_add_input = GetOpWithOutput(*model, add_op->inputs[1 - i]);
125 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]);
|
| unpartition_embedding_lookup.cc | 71 auto* op = GetOpWithOutput(*model, indices_partition_output_name);
116 auto* op = GetOpWithOutput(*model, gather_output_name);
131 auto* op = GetOpWithOutput(*model, gather_op->inputs[1]);
164 Operator* div_op = GetOpWithOutput(*model, data_partition_op->inputs[0]);
165 Operator* mod_op = GetOpWithOutput(*model, data_partition_op->inputs[1]);
170 CHECK_EQ(mod_op, GetOpWithOutput(*model, indices_partition_op->inputs[1]))
|
| identify_l2_pool.cc | 56 Operator* prev_to_sqrt_op = GetOpWithOutput(*model, sqrt_op->inputs[0]);
75 square_op = GetOpWithOutput(*model, avpool_op->inputs[0]);
|
| remove_trivial_fake_quant.cc | 42 auto* producing_op = GetOpWithOutput(model, fakequant_op.inputs[0]);
|
| remove_unused_op.cc | 96 if (!GetOpWithOutput(*model, input)) {
|
| fuse_activation_functions.cc | 42 Operator* op = GetOpWithOutput(*model, ac_op->inputs[0]);
|
| make_initial_dequantize_operator.cc | 41 if (GetOpWithOutput(*model, input_name)) {
|
| resolve_tensorflow_switch.cc | 118 !GetOpWithOutput(*model, input)) {
|
| group_bidirectional_sequence_ops.cc | 43 *fw_output = GetOpWithOutput(model, op.inputs[0]);
44 *bw_output = GetOpWithOutput(model, op.inputs[1]);
68 auto* reverse_output = GetOpWithOutput(model, op->inputs[1]);
88 op_it = GetOpWithOutput(model, output_op.inputs[0]);
96 op_it = GetOpWithOutput(model, op_it->inputs[0]);
164 GetOpWithOutput(model, first_bw_sequence_op_input->inputs[0]);
|
| convert_expanddims_to_reshape.cc | 86 !GetOpWithOutput(*model, axis_array_name)) {
|
| propagate_activation_function_into_constants.cc | 41 auto* src_op = GetOpWithOutput(*model, ac_op->inputs[0]);
|
| convert_reorder_axes.cc | 109 const auto* op_producing_input = GetOpWithOutput(*model, input_array_name);
|
| resolve_tensorflow_matmul.cc | 180 !GetOpWithOutput(*model, previous_op_shape)) {
|
| dequantize.cc | 112 auto* op_outputting_array = GetOpWithOutput(*model, array_name);
|
| resolve_constant_concatenation.cc | 154 const Operator* input_op = GetOpWithOutput(*model, input_name);
|
| identify_lstm.cc | 44 *source_op = GetOpWithOutput(model, array_name);
|
| fuse_binary_into_preceding_affine.cc | 241 GetOpWithOutput(*model, binary_op->inputs[index_of_variable_input]);
|
| quantize.cc | 508 const auto* other_op = GetOpWithOutput(*model, input);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| tooling_util.h | 79 Operator* GetOpWithOutput(const Model& model, const string& array_name);
|
| tooling_util.cc | 209 Operator* GetOpWithOutput(const Model& model, const string& array_name) {
912 QCHECK(GetOpWithOutput(model, output_array))
923 QCHECK(GetOpWithOutput(model, rnn_state.back_edge_source_array()))
[all...] |
| dump_graphviz.cc | 543 GetOpWithOutput(model, input) == nullptr) {
|
| import_tensorflow.cc | 635 GetOpWithOutput(*model, reordered_weights_name);
711 GetOpWithOutput(*model, reordered_weights_name);
[all...] |
| export_tensorflow.cc | 375 const auto* op = GetOpWithOutput(model, name);
[all...] |
