Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc87 const std::string& back_edge_source_array, in GetMatchingRnnArray() argument
90 if (rnn_state.back_edge_source_array() == back_edge_source_array) { in GetMatchingRnnArray()
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()
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()
Dlstm_utils.h107 const std::string& back_edge_source_array,
Ddequantize.cc145 if (array_name == rnn_state.back_edge_source_array()) { in DequantizeArray()
Dresolve_constant_unary.cc149 if (unary_op->inputs[0] == rnn_state.back_edge_source_array()) { in Run()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc125 if (rnn_state.back_edge_source_array() == name) { in IsArrayConsumed()
952 QCHECK(GetOpWithOutput(model, rnn_state.back_edge_source_array())) in CheckNonExistentIOArrays()
954 << rnn_state.back_edge_source_array() in CheckNonExistentIOArrays()
994 model->GetOrCreateArray(rnn_state.back_edge_source_array()); in FixNoMissingArray()
1016 arrays_without_known_use.erase(rnn_state.back_edge_source_array()); in CheckNoOrphanedArray()
1041 arrays_without_known_use.erase(rnn_state.back_edge_source_array()); in FixNoOrphanedArray()
2239 if (array_name == rnn_state.back_edge_source_array()) { in IsDiscardableArray()
2337 if (!model->HasArray(rnn_state.back_edge_source_array()) || in FinishBuildingRNNStates()
2339 CHECK(model->HasArray(rnn_state.back_edge_source_array())); in FinishBuildingRNNStates()
2343 const auto& src_array = model->GetArray(rnn_state.back_edge_source_array()); in FinishBuildingRNNStates()
Ddump_graphviz.cc139 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayColorAndShape()
184 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayCompassPt()
779 AppendF(output_file, kRNNBackEdgeFmt, rnn_state.back_edge_source_array(), in DumpGraphviz()
Dmodel_flags.proto89 optional string back_edge_source_array = 2; field
Dimport_tensorflow.cc2226 consumed_arrays.push_back(rnn_state.back_edge_source_array()); in AddExtraOutputs()