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.cc86 bool GetMatchingRnnArray(Model* model, const string& back_edge_source_array, in GetMatchingRnnArray() argument
89 if (rnn_state.back_edge_source_array() == back_edge_source_array) { in GetMatchingRnnArray()
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()
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()
Dlstm_utils.h106 bool GetMatchingRnnArray(Model* model, const 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()
923 QCHECK(GetOpWithOutput(model, rnn_state.back_edge_source_array())) in CheckNonExistentIOArrays()
925 << rnn_state.back_edge_source_array() in CheckNonExistentIOArrays()
965 model->GetOrCreateArray(rnn_state.back_edge_source_array()); in FixNoMissingArray()
987 arrays_without_known_use.erase(rnn_state.back_edge_source_array()); in CheckNoOrphanedArray()
1012 arrays_without_known_use.erase(rnn_state.back_edge_source_array()); in FixNoOrphanedArray()
2195 if (array_name == rnn_state.back_edge_source_array()) { in IsDiscardableArray()
2281 if (!model->HasArray(rnn_state.back_edge_source_array()) || in FinishBuildingRNNStates()
2283 CHECK(model->HasArray(rnn_state.back_edge_source_array())); in FinishBuildingRNNStates()
2287 const auto& src_array = model->GetArray(rnn_state.back_edge_source_array()); in FinishBuildingRNNStates()
Ddump_graphviz.cc137 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayColorAndShape()
181 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayCompassPt()
774 AppendF(output_file, kRNNBackEdgeFmt, rnn_state.back_edge_source_array(), in DumpGraphviz()
Dmodel_flags.proto82 optional string back_edge_source_array = 2; field
Dimport_tensorflow.cc2098 consumed_arrays.push_back(rnn_state.back_edge_source_array()); in AddExtraOutputs()