Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_reorder_axes.cc92 const std::string input_array_name = reorder_op->inputs[0]; in Run() local
95 auto& input_array = model->GetArray(input_array_name); in Run()
120 AddMessageF("Reordered axes for array %s", input_array_name); in Run()
123 RenameArray(model, output_array_name, input_array_name); in Run()
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()
Dgroup_bidirectional_sequence_ops.cc200 const std::string& input_array_name = in ConstructBidirectionalSequenceOp() local
202 model->GetOrCreateArray(input_array_name); in ConstructBidirectionalSequenceOp()
204 (*bi_op)->inputs.push_back(input_array_name); in ConstructBidirectionalSequenceOp()
263 const std::string& input_array_name = in ConstructBidirectionalSequenceOp() local
265 model->GetOrCreateArray(input_array_name); in ConstructBidirectionalSequenceOp()
267 (*bi_op)->inputs.push_back(input_array_name); in ConstructBidirectionalSequenceOp()
321 OperatorType operator_type, const std::string& input_array_name, in RewireBidirectionalSequenceSequenceOpsConnections() argument
336 std::string cur_fw_input = input_array_name; in RewireBidirectionalSequenceSequenceOpsConnections()
337 std::string cur_bw_input = input_array_name; in RewireBidirectionalSequenceSequenceOpsConnections()
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()
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()