Home
last modified time | relevance | path

Searched refs:back_edge_source_array (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dlstm_utils.cc89 const std::string& back_edge_source_array, in GetMatchingRnnArray() argument
92 if (rnn_state.back_edge_source_array() == back_edge_source_array) { in GetMatchingRnnArray()
H A Dgraph_transformations.cc88 !useful_arrays.count(rnn_state.back_edge_source_array()); in DiscardUselessConnectedComponentsAndRNNBackEdges()
89 useful_arrays.insert(rnn_state.back_edge_source_array()); in DiscardUselessConnectedComponentsAndRNNBackEdges()
116 !model->HasArray(rnn_state.back_edge_source_array()) || in DiscardUselessConnectedComponentsAndRNNBackEdges()
H A Dremove_unused_op.cc67 if (output == rnn_state.back_edge_source_array()) { in Run()
69 if (!IsDiscardableArray(*model, rnn_state.back_edge_source_array()) || in Run()
H A Dlstm_utils.h107 const std::string& back_edge_source_array,
H A Ddequantize.cc145 if (array_name == rnn_state.back_edge_source_array()) { in DequantizeArray()
H A Dresolve_constant_unary.cc151 if (unary_op->inputs[0] == rnn_state.back_edge_source_array()) { in Run()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc127 if (rnn_state.back_edge_source_array() == name) { in IsArrayConsumed()
954 QCHECK(GetOpWithOutput(model, rnn_state.back_edge_source_array())) in CheckNonExistentIOArrays()
956 << rnn_state.back_edge_source_array() in CheckNonExistentIOArrays()
996 model->GetOrCreateArray(rnn_state.back_edge_source_array()); in FixNoMissingArray()
1018 arrays_without_known_use.erase(rnn_state.back_edge_source_array()); in CheckNoOrphanedArray()
1043 arrays_without_known_use.erase(rnn_state.back_edge_source_array()); in FixNoOrphanedArray()
2241 if (array_name == rnn_state.back_edge_source_array()) { in IsDiscardableArray()
2344 if (!model->HasArray(rnn_state.back_edge_source_array()) || in FinishBuildingRNNStates()
2346 CHECK(model->HasArray(rnn_state.back_edge_source_array())); in FinishBuildingRNNStates()
2350 const auto& src_array = model->GetArray(rnn_state.back_edge_source_array()); in FinishBuildingRNNStates()
H A Ddump_graphviz.cc141 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayColorAndShape()
186 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayCompassPt()
781 AppendF(output_file, kRNNBackEdgeFmt, rnn_state.back_edge_source_array(), in DumpGraphviz()
H A Dmodel_flags.proto89 optional string back_edge_source_array = 2; field
H A Dimport_tensorflow.cc2233 consumed_arrays.push_back(rnn_state.back_edge_source_array()); in AddExtraOutputs()