| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_l2_normalization.cc | 58 GetOpWithOutput(*model, div_or_mul_op->inputs[0]), in Run()
59 GetOpWithOutput(*model, div_or_mul_op->inputs[1]), in Run()
69 GetOpWithOutput(*model, sqrt_or_rsqrt_op->inputs[0]); in Run()
102 op_producing_add_input = GetOpWithOutput(*model, add_op->inputs[1 - i]); in Run()
125 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]); in Run()
|
| D | unpartition_embedding_lookup.cc | 71 auto* op = GetOpWithOutput(*model, indices_partition_output_name); in Run()
116 auto* op = GetOpWithOutput(*model, gather_output_name); in Run()
131 auto* op = GetOpWithOutput(*model, gather_op->inputs[1]); in Run()
164 Operator* div_op = GetOpWithOutput(*model, data_partition_op->inputs[0]); in Run()
165 Operator* mod_op = GetOpWithOutput(*model, data_partition_op->inputs[1]); in Run()
170 CHECK_EQ(mod_op, GetOpWithOutput(*model, indices_partition_op->inputs[1])) in Run()
|
| D | identify_prelu.cc | 57 const auto* relu_input_op = GetOpWithOutput(*model, add_op->inputs[0]); in Run()
67 const auto* mul_op = GetOpWithOutput(*model, add_op->inputs[1]); in Run()
76 const auto* relu_neg_input_op = GetOpWithOutput(*model, mul_op->inputs[1]); in Run()
92 GetOpWithOutput(*model, relu_neg_input_op->inputs[0]); in Run()
|
| D | fuse_broadcast_into_following_binary.cc | 74 GetOpWithOutput(*model, binary_op->inputs[0]), in Run()
75 GetOpWithOutput(*model, binary_op->inputs[1]), in Run()
|
| D | identify_l2_pool.cc | 56 Operator* prev_to_sqrt_op = GetOpWithOutput(*model, sqrt_op->inputs[0]); in Run()
75 square_op = GetOpWithOutput(*model, avpool_op->inputs[0]); in Run()
|
| D | group_bidirectional_sequence_ops.cc | 43 *fw_output = GetOpWithOutput(model, op.inputs[0]); in MatchTwoUnpackOps()
44 *bw_output = GetOpWithOutput(model, op.inputs[1]); in MatchTwoUnpackOps()
68 auto* reverse_output = GetOpWithOutput(model, op->inputs[1]); in MatchDynamicBidirectionalSequenceOutputs()
88 op_it = GetOpWithOutput(model, output_op.inputs[0]); in FindUnidirectionalSequenceOp()
96 op_it = GetOpWithOutput(model, op_it->inputs[0]); in FindUnidirectionalSequenceOp()
164 GetOpWithOutput(model, first_bw_sequence_op_input->inputs[0]); in CheckTwoUnidirectionalSequenceOpsAreValid()
|
| D | remove_trivial_fake_quant.cc | 42 auto* producing_op = GetOpWithOutput(model, fakequant_op.inputs[0]); in IsFakeQuantTrivial()
|
| D | remove_unused_op.cc | 96 if (!GetOpWithOutput(*model, input)) { in Run()
|
| D | make_initial_dequantize_operator.cc | 41 if (GetOpWithOutput(*model, input_name)) { in AddDequantizeOperatorToInput()
|
| D | convert_expanddims_to_reshape.cc | 86 !GetOpWithOutput(*model, axis_array_name)) { in Run()
|
| D | resolve_tensorflow_switch.cc | 118 !GetOpWithOutput(*model, input)) { in Run()
|
| D | fuse_activation_functions.cc | 42 Operator* op = GetOpWithOutput(*model, ac_op->inputs[0]); in Run()
|
| D | propagate_activation_function_into_constants.cc | 41 auto* src_op = GetOpWithOutput(*model, ac_op->inputs[0]); in Run()
|
| D | convert_reorder_axes.cc | 109 const auto* op_producing_input = GetOpWithOutput(*model, input_array_name); in Run()
|
| D | resolve_tensorflow_matmul.cc | 180 !GetOpWithOutput(*model, previous_op_shape)) { in Run()
|
| D | dequantize.cc | 112 auto* op_outputting_array = GetOpWithOutput(*model, array_name); in DequantizeArray()
|
| D | resolve_constant_concatenation.cc | 154 const Operator* input_op = GetOpWithOutput(*model, input_name); in Run()
|
| D | identify_lstm.cc | 44 *source_op = GetOpWithOutput(model, array_name); in ValidateSourceOp()
|
| D | fuse_binary_into_preceding_affine.cc | 241 GetOpWithOutput(*model, binary_op->inputs[index_of_variable_input]); in Run()
|
| D | quantize.cc | 508 const auto* other_op = GetOpWithOutput(*model, input); in Run()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 209 Operator* GetOpWithOutput(const Model& model, const string& array_name) { in GetOpWithOutput() function
912 QCHECK(GetOpWithOutput(model, output_array)) in CheckNonExistentIOArrays()
923 QCHECK(GetOpWithOutput(model, rnn_state.back_edge_source_array())) in CheckNonExistentIOArrays()
1060 if (!GetOpWithOutput(model, array_entry.first)) { in CheckOperatorOrdering()
1085 if (!GetOpWithOutput(*model, array_entry.first)) { in FixOperatorOrdering()
1259 if (!GetOpWithOutput(*model, output_array_name)) { in FixEdgeArrays()
|
| D | tooling_util.h | 79 Operator* GetOpWithOutput(const Model& model, const string& array_name);
|
| D | dump_graphviz.cc | 543 GetOpWithOutput(model, input) == nullptr) { in DumpOperatorEdges()
|
| D | import_tensorflow.cc | 635 GetOpWithOutput(*model, reordered_weights_name); in ConvertConvOperator()
711 GetOpWithOutput(*model, reordered_weights_name); in ConvertDepthwiseConvOperator()
1889 GetOpWithOutput(*model, transposed_weights_name); in ConvertTransposeConvOperator()
2116 auto* producer_op = GetOpWithOutput(*model, split[0]); in AddExtraOutputs()
|
| D | export_tensorflow.cc | 375 const auto* op = GetOpWithOutput(model, name); in WalkUpToConstantArray()
812 Operator* providing_op = GetOpWithOutput(model, src_op.inputs[0]); in ConvertSoftmaxOperator()
852 Operator* providing_op = GetOpWithOutput(model, src_op.inputs[0]); in ConvertLogSoftmaxOperator()
|