Searched refs:output_array_name (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_reorder_axes.cc | 93 const string output_array_name = reorder_op->outputs[0]; in Run() local
96 auto& output_array = model->GetArray(output_array_name); in Run()
123 RenameArray(model, output_array_name, input_array_name); in Run()
|
| 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()
|
| D | remove_trivial_quantized_activation_func.cc | 63 const string& output_array_name) { in IsTrivialFusedActivationFunc() argument
87 const auto& output_array = model.GetArray(output_array_name); in IsTrivialFusedActivationFunc()
|
| D | convert_reorder_axes.cc | 101 const auto& output_array_name = reorder_op->outputs[0]; in Run() local
103 auto& output_array = model->GetArray(output_array_name); in Run()
|
| D | group_bidirectional_sequence_ops.cc | 372 const string& output_array_name = original_unpack_operator.outputs[i]; in RewireFinalUnpackOutputs() local
377 Operator* unpack_following_op = GetOpWithInput(*model, output_array_name); in RewireFinalUnpackOutputs()
392 for (const string& output_array_name : unpack_op.outputs) { in RemoveUnpackOperator() local
393 DeleteArrayIfUnused(output_array_name, model); in RemoveUnpackOperator()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 1258 for (const string& output_array_name : model->flags.output_arrays()) { in FixEdgeArrays() local
1259 if (!GetOpWithOutput(*model, output_array_name)) { in FixEdgeArrays()
1261 LOG(WARNING) << "Fixing constant output array " << output_array_name in FixEdgeArrays()
1264 AvailableArrayName(*model, output_array_name + "_copy"); in FixEdgeArrays()
1265 CloneArray(model, output_array_name, intermediate_array_name); in FixEdgeArrays()
1266 InsertCopyOperator(model, intermediate_array_name, output_array_name); in FixEdgeArrays()
|