| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_l2_normalization.cc | 44 GetOpWithOutput(*model, div_or_mul_op->inputs[0]), in Run()
45 GetOpWithOutput(*model, div_or_mul_op->inputs[1]), in Run()
55 GetOpWithOutput(*model, sqrt_or_rsqrt_op->inputs[0]); in Run()
88 op_producing_add_input = GetOpWithOutput(*model, add_op->inputs[1 - i]); in Run()
111 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]); 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 | unpartition_embedding_lookup.cc | 73 auto* op = GetOpWithOutput(*model, indices_partition_output_name); in Run()
118 auto* op = GetOpWithOutput(*model, gather_output_name); in Run()
133 auto* op = GetOpWithOutput(*model, gather_op->inputs[1]); in Run()
166 Operator* div_op = GetOpWithOutput(*model, data_partition_op->inputs[0]); in Run()
167 Operator* mod_op = GetOpWithOutput(*model, data_partition_op->inputs[1]); in Run()
172 CHECK_EQ(mod_op, GetOpWithOutput(*model, indices_partition_op->inputs[1])) 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 | 42 Operator* prev_to_sqrt_op = GetOpWithOutput(*model, sqrt_op->inputs[0]); in Run()
61 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 | make_initial_dequantize_operator.cc | 42 if (GetOpWithOutput(*model, input_name)) { in AddDequantizeOperatorToInput()
|
| D | resolve_tensorflow_switch.cc | 127 !GetOpWithOutput(*model, input)) { in Run()
|
| D | fuse_activation_functions.cc | 42 Operator* op = GetOpWithOutput(*model, ac_op->inputs[0]); in Run()
|
| D | convert_expanddims_to_reshape.cc | 93 !GetOpWithOutput(*model, axis_array_name)) { 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 | 110 const auto* op_producing_input = GetOpWithOutput(*model, input_array_name); in Run()
|
| D | identify_dilated_conv.cc | 129 auto* pre_stb_pad_op = GetOpWithOutput(*model, stb_op->inputs[0]); in ResolveDilatedConv()
|
| D | resolve_constant_concatenation.cc | 154 const Operator* input_op = GetOpWithOutput(*model, input_name); in Run()
|
| D | dequantize.cc | 112 auto* op_outputting_array = GetOpWithOutput(*model, array_name); in DequantizeArray()
|
| D | identify_nearest_upsample.cc | 96 Operator* next_op = GetOpWithOutput(*model, output); in Run()
|
| D | identify_lstm.cc | 44 *source_op = GetOpWithOutput(model, array_name); in ValidateSourceOp()
|
| D | fuse_binary_into_preceding_affine.cc | 256 GetOpWithOutput(*model, binary_op->inputs[index_of_variable_input]); in Run()
|
| D | quantize.cc | 557 const auto* other_op = GetOpWithOutput(*model, input); in Run()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 161 GetOpWithOutput(*model, array_name) == nullptr) { in DeleteArrayIfUnused()
181 GetOpWithOutput(*model, array_name); in DeleteArrayIfUnusedOutsideOfOp()
225 Operator* GetOpWithOutput(const Model& model, const std::string& array_name) { in GetOpWithOutput() function
941 QCHECK(GetOpWithOutput(model, output_array)) in CheckNonExistentIOArrays()
952 QCHECK(GetOpWithOutput(model, rnn_state.back_edge_source_array())) in CheckNonExistentIOArrays()
1091 if (!GetOpWithOutput(model, array_entry.first)) { in CheckOperatorOrdering()
1116 if (!GetOpWithOutput(*model, array_entry.first)) { in FixOperatorOrdering()
1290 if (!GetOpWithOutput(*model, output_array_name)) { in FixEdgeArrays()
|
| D | tooling_util.h | 75 Operator* GetOpWithOutput(const Model& model, const std::string& array_name);
|
| D | import_tensorflow.cc | 847 GetOpWithOutput(*model, reordered_weights_name); in ConvertConvOperator()
929 GetOpWithOutput(*model, reordered_weights_name); in ConvertDepthwiseConvOperator()
2014 GetOpWithOutput(*model, transposed_weights_name); in ConvertTransposeConvOperator()
2234 if (GetOpWithOutput(*model, consumed_array)) { in AddExtraOutputs()
2250 auto* producer_op = GetOpWithOutput(*model, split[0]); in AddExtraOutputs()
|
| D | dump_graphviz.cc | 548 GetOpWithOutput(model, input) == nullptr) { in DumpOperatorEdges()
|
| D | export_tensorflow.cc | 379 const auto* op = GetOpWithOutput(model, name); in WalkUpToConstantArray()
817 Operator* providing_op = GetOpWithOutput(model, src_op.inputs[0]); in ConvertSoftmaxOperator()
858 Operator* providing_op = GetOpWithOutput(model, src_op.inputs[0]); in ConvertLogSoftmaxOperator()
|