Home
last modified time | relevance | path

Searched refs:output_state (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_test.py134 def _AddUp(self, outputs, output_state): argument
136 for s in output_state:
153 def output_state(self): member in CudnnTestModel
162 return self._AddUp(self.outputs, self.output_state)
211 outputs, output_state = self._rnn(
213 return self._AddUp(outputs, output_state)
227 [self.outputs, self.output_state],
988 output_state = array_ops.stack([s for s in states])
1004 output_state = []
1006 output_state.append(array_ops.stack([s_fw, s_bw]))
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h84 TfLiteTensor* cell_state_quantized, TfLiteTensor* output_state,
Dlstm_eval.cc1097 TfLiteTensor* cell_state_quantized, TfLiteTensor* output_state, in EvalHybrid()
1313 quantized_cell_state_ptr, output_state->data.f, cell_state->data.f, in EvalHybrid()
1334 output_state->data.f + b * output_batch_leading_dim; in EvalHybrid()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc730 auto& output_state = node_map_[output_node]; in AddOutputNodesToReadyQueue() local
731 output_state.num_inputs_ready++; in AddOutputNodesToReadyQueue()
735 if (output_state.num_inputs_ready == output_state.inputs.size() || in AddOutputNodesToReadyQueue()
738 output_state.time_ready = curr_time; in AddOutputNodesToReadyQueue()