HomeSort by relevance Sort by last modified time
    Searched defs:output_state (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
cudnn_rnn_test.py 134 def _AddUp(self, outputs, output_state):
136 for s in output_state:
153 def output_state(self): member in class: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],
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler.cc 730 auto& output_state = node_map_[output_node]; local
731 output_state.num_inputs_ready++;
735 if (output_state.num_inputs_ready == output_state.inputs.size() ||
738 output_state.time_ready = curr_time;
    [all...]

Completed in 761 milliseconds