Searched refs:output_array_name (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | resolve_reorder_axes.cc | 94 const std::string output_array_name = reorder_op->outputs[0]; in Run() local
97 auto& output_array = model->GetArray(output_array_name); in Run()
124 RenameArray(model, output_array_name, input_array_name); in Run()
|
| H A D | resolve_multiply_by_zero.cc | 63 const auto& output_array_name = mul_op->outputs[0]; in Run() local
64 auto& output_array = model->GetArray(output_array_name); in Run()
66 if (!IsDiscardableArray(*model, output_array_name)) { in Run()
|
| H A D | remove_trivial_quantized_activation_func.cc | 63 const std::string& output_array_name) { in IsTrivialFusedActivationFunc() argument
87 const auto& output_array = model.GetArray(output_array_name); in IsTrivialFusedActivationFunc()
|
| H A D | convert_reorder_axes.cc | 102 const auto& output_array_name = reorder_op->outputs[0]; in Run() local
104 auto& output_array = model->GetArray(output_array_name); in Run()
|
| H A D | group_bidirectional_sequence_ops.cc | 375 const std::string& output_array_name = original_unpack_operator.outputs[i]; in RewireFinalUnpackOutputs() local
380 Operator* unpack_following_op = GetOpWithInput(*model, output_array_name); in RewireFinalUnpackOutputs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | tooling_util.cc | 1291 for (const std::string& output_array_name : model->flags.output_arrays()) { in FixEdgeArrays() local
1292 if (!GetOpWithOutput(*model, output_array_name)) { in FixEdgeArrays()
1294 LOG(WARNING) << "Fixing constant output array " << output_array_name in FixEdgeArrays()
1297 AvailableArrayName(*model, output_array_name + "_copy"); in FixEdgeArrays()
1298 CloneArray(model, output_array_name, intermediate_array_name); in FixEdgeArrays()
1299 InsertCopyOperator(model, intermediate_array_name, output_array_name); in FixEdgeArrays()
|