Home
last modified time | relevance | path

Searched refs:FindOpWithOutput (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dreorder_elementwise_unary.cc79 auto it = FindOpWithOutput(*model, intermediate_name); in Run()
Dmerge_reshape_into_preceding_transpose.cc129 const auto& transpose_it = FindOpWithOutput(*model, intermediate_name); in Run()
Dmove_binary_operator_before_reshape.cc115 FindOpWithOutput(*model, binary_op->inputs[variable_input_idx]); in Run()
Dreorder_reshape_transpose.cc123 auto reshape_it = FindOpWithOutput(*model, transpose_op->inputs[0]); in Run()
Dquantize.cc586 auto dequantize_it = FindOpWithOutput(*model, input); in Run()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h73 std::vector<std::unique_ptr<Operator>>::const_iterator FindOpWithOutput(
77 std::vector<std::unique_ptr<Operator>>::iterator FindOpWithOutput(
Dtooling_util.cc201 std::vector<std::unique_ptr<Operator>>::const_iterator FindOpWithOutput( in FindOpWithOutput() function
213 std::vector<std::unique_ptr<Operator>>::iterator FindOpWithOutput( in FindOpWithOutput() function
226 auto it = FindOpWithOutput(model, array_name); in GetOpWithOutput()