Home
last modified time | relevance | path

Searched refs:input_array_name (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dgroup_bidirectional_sequence_ops.cc201 const std::string& input_array_name = in ConstructBidirectionalSequenceOp() local
203 model->GetOrCreateArray(input_array_name); in ConstructBidirectionalSequenceOp()
205 (*bi_op)->inputs.push_back(input_array_name); in ConstructBidirectionalSequenceOp()
264 const std::string& input_array_name = in ConstructBidirectionalSequenceOp() local
266 model->GetOrCreateArray(input_array_name); in ConstructBidirectionalSequenceOp()
268 (*bi_op)->inputs.push_back(input_array_name); in ConstructBidirectionalSequenceOp()
322 OperatorType operator_type, const std::string& input_array_name, in RewireBidirectionalSequenceSequenceOpsConnections() argument
337 std::string cur_fw_input = input_array_name; in RewireBidirectionalSequenceSequenceOpsConnections()
338 std::string cur_bw_input = input_array_name; in RewireBidirectionalSequenceSequenceOpsConnections()
H A Dresolve_reorder_axes.cc93 const std::string input_array_name = reorder_op->inputs[0]; in Run() local
96 auto& input_array = model->GetArray(input_array_name); in Run()
121 AddMessageF("Reordered axes for array %s", input_array_name); in Run()
124 RenameArray(model, output_array_name, input_array_name); in Run()
H A Dconvert_reorder_axes.cc101 const auto& input_array_name = reorder_op->inputs[0]; in Run() local
103 auto& input_array = model->GetArray(input_array_name); in Run()
108 std::string constant_input_array_name = input_array_name; in Run()
110 const auto* op_producing_input = GetOpWithOutput(*model, input_array_name); in Run()
H A Dremove_trivial_quantized_min_max.cc34 OperatorType op_type, const std::string& input_array_name, in IsTrivialMinMax() argument
63 const auto& input_array = model.GetArray(input_array_name); in IsTrivialMinMax()
H A Dremove_trivial_quantized_activation_func.cc35 const std::string& input_array_name) { in IsTrivialUnfusedActivationFunc() argument
55 const auto& input_array = model.GetArray(input_array_name); in IsTrivialUnfusedActivationFunc()