Home
last modified time | relevance | path

Searched refs:rnn_state (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dremove_unused_op.cc50 for (const auto& rnn_state : model->flags.rnn_states()) { in Run() local
51 if (output == rnn_state.state_array()) { in Run()
66 for (const auto& rnn_state : model->flags.rnn_states()) { in Run() local
67 if (output == rnn_state.back_edge_source_array()) { in Run()
69 if (!IsDiscardableArray(*model, rnn_state.back_edge_source_array()) || in Run()
70 !IsDiscardableArray(*model, rnn_state.state_array()) || in Run()
71 CountOpsWithInput(*model, rnn_state.state_array())) { in Run()
Dgraph_transformations.cc83 for (const auto& rnn_state : model->flags.rnn_states()) { in DiscardUselessConnectedComponentsAndRNNBackEdges() local
85 useful_arrays.count(rnn_state.state_array()); in DiscardUselessConnectedComponentsAndRNNBackEdges()
88 !useful_arrays.count(rnn_state.back_edge_source_array()); in DiscardUselessConnectedComponentsAndRNNBackEdges()
89 useful_arrays.insert(rnn_state.back_edge_source_array()); in DiscardUselessConnectedComponentsAndRNNBackEdges()
114 for (const auto& rnn_state : model->flags.rnn_states()) { in DiscardUselessConnectedComponentsAndRNNBackEdges() local
116 !model->HasArray(rnn_state.back_edge_source_array()) || in DiscardUselessConnectedComponentsAndRNNBackEdges()
117 !model->HasArray(rnn_state.state_array()); in DiscardUselessConnectedComponentsAndRNNBackEdges()
119 CHECK(rnn_state.discardable()); in DiscardUselessConnectedComponentsAndRNNBackEdges()
121 rnn_states_to_keep.push_back(rnn_state); in DiscardUselessConnectedComponentsAndRNNBackEdges()
125 for (const auto& rnn_state : rnn_states_to_keep) { in DiscardUselessConnectedComponentsAndRNNBackEdges() local
[all …]
Dlstm_utils.cc89 for (const auto& rnn_state : model->flags.rnn_states()) { in GetMatchingRnnArray() local
90 if (rnn_state.back_edge_source_array() == back_edge_source_array) { in GetMatchingRnnArray()
91 *rnn_array = rnn_state.state_array(); in GetMatchingRnnArray()
Ddequantize.cc141 for (const auto& rnn_state : model->flags.rnn_states()) { in DequantizeArray() local
142 if (array_name == rnn_state.state_array()) { in DequantizeArray()
145 if (array_name == rnn_state.back_edge_source_array()) { in DequantizeArray()
Dresolve_constant_unary.cc148 for (const auto& rnn_state : model->flags.rnn_states()) { in Run() local
149 if (unary_op->inputs[0] == rnn_state.back_edge_source_array()) { in Run()
152 if (unary_op->inputs[0] == rnn_state.state_array()) { in Run()
Dquantize.cc646 for (const auto& rnn_state : model->flags.rnn_states()) { in Run() local
647 if (rnn_state.state_array() == input) { in Run()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc124 for (const auto& rnn_state : model.flags.rnn_states()) { in IsArrayConsumed() local
125 if (rnn_state.back_edge_source_array() == name) { in IsArrayConsumed()
945 for (const auto& rnn_state : model.flags.rnn_states()) { in CheckNonExistentIOArrays() local
946 if (!rnn_state.discardable()) { in CheckNonExistentIOArrays()
948 QCHECK(GetOpWithInput(model, rnn_state.state_array())) in CheckNonExistentIOArrays()
949 << "Specified RNN state \"" << rnn_state.state_array() in CheckNonExistentIOArrays()
952 QCHECK(GetOpWithOutput(model, rnn_state.back_edge_source_array())) in CheckNonExistentIOArrays()
954 << rnn_state.back_edge_source_array() in CheckNonExistentIOArrays()
992 for (const auto& rnn_state : model->flags.rnn_states()) { in FixNoMissingArray() local
993 model->GetOrCreateArray(rnn_state.state_array()); in FixNoMissingArray()
[all …]
Ddump_graphviz.cc130 for (const auto& rnn_state : model.flags.rnn_states()) { in GetArrayColorAndShape() local
132 if (array_name == rnn_state.state_array()) { in GetArrayColorAndShape()
139 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayColorAndShape()
178 for (const auto& rnn_state : model.flags.rnn_states()) { in GetArrayCompassPt() local
180 if (array_name == rnn_state.state_array()) { in GetArrayCompassPt()
184 if (array_name == rnn_state.back_edge_source_array()) { in GetArrayCompassPt()
778 for (const auto& rnn_state : model.flags.rnn_states()) { in DumpGraphviz() local
779 AppendF(output_file, kRNNBackEdgeFmt, rnn_state.back_edge_source_array(), in DumpGraphviz()
780 rnn_state.state_array()); in DumpGraphviz()
Dallocate_transient_arrays.cc79 for (const auto& rnn_state : model.flags.rnn_states()) { in ComputeArrayLifespans() local
82 (*array_lifespans)[rnn_state.state_array()] = lifespan; in ComputeArrayLifespans()
173 for (const auto& rnn_state : model.flags.rnn_states()) { in TransientArraySize() local
174 if (rnn_state.state_array() == array_name) { in TransientArraySize()
Dimport_tensorflow.cc2128 auto* rnn_state = model->flags.add_rnn_states(); in ConvertOperatorSpecialCasedAsRNNBackEdge() local
2131 rnn_state->set_discardable(true); in ConvertOperatorSpecialCasedAsRNNBackEdge()
2132 rnn_state->set_state_array(node.name()); in ConvertOperatorSpecialCasedAsRNNBackEdge()
2133 rnn_state->set_back_edge_source_array(node.input(0)); in ConvertOperatorSpecialCasedAsRNNBackEdge()
2136 rnn_state->set_size(1); in ConvertOperatorSpecialCasedAsRNNBackEdge()
2225 for (const auto& rnn_state : model->flags.rnn_states()) { in AddExtraOutputs() local
2226 consumed_arrays.push_back(rnn_state.back_edge_source_array()); in AddExtraOutputs()
2781 for (const auto& rnn_state : model->flags.rnn_states()) { in ImportTensorFlowGraphDef() local
2782 model->GetArray(rnn_state.state_array()).buffer = nullptr; in ImportTensorFlowGraphDef()
Dexport_tensorflow.cc2490 for (const auto& rnn_state : model.flags.rnn_states()) { in ExportTensorFlowGraphDefImplementation() local
2491 AddPlaceholderForRNNState(model, rnn_state.state_array(), rnn_state.size(), in ExportTensorFlowGraphDefImplementation()
/external/libopus/src/
Danalysis.h78 float rnn_state[MAX_NEURONS]; member
Danalysis.c933 compute_gru(&layer1, tonal->rnn_state, layer_out); in tonality_analysis()
934 compute_dense(&layer2, frame_probs, tonal->rnn_state); in tonality_analysis()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc1088 RnnScratchSpace& rnn_state = (*cache)[key]; in GetCachedRnnDescriptor() local
1089 if (rnn_state.rnn_desc == nullptr || ResetRndGenState()) { in GetCachedRnnDescriptor()
1092 rnn_state.dropout_state_allocator.reset(dropout_state_allocator); in GetCachedRnnDescriptor()
1095 dropout_state_allocator, &rnn_state.rnn_desc, use_padded_io); in GetCachedRnnDescriptor()
1098 *rnn_desc = rnn_state.rnn_desc.get(); in GetCachedRnnDescriptor()