HomeSort by relevance Sort by last modified time
    Searched refs:state_shape (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/
convolutional_recurrent.py 379 state_shape = [None] * 4
380 state_shape[channel_axis] = input_dim
381 state_shape = tuple(state_shape)
383 InputSpec(shape=state_shape),
384 InputSpec(shape=state_shape)
wrappers.py 312 state_shape = output_shape[1:]
324 return output_shape + state_shape + copy.copy(state_shape)
325 return [output_shape] + state_shape + copy.copy(state_shape)
recurrent.py 445 state_shape = [(input_shape[0], dim) for dim in state_size]
446 return [output_shape] + state_shape
    [all...]
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
cudnn_rnn.py 409 def state_shape(self, batch_size): member in class:_CudnnRNN
414 for sp in self.state_shape(batch_size):
511 def state_shape(self, batch_size): member in class:CudnnLSTM
528 def state_shape(self, batch_size): member in class:_CudnnRNNNoInputC
  /external/tensorflow/tensorflow/contrib/bayesflow/python/kernel_tests/
mcmc_diagnostics_test.py 307 state_shape = sample_shape + independent_chain_shape + other_shape
308 state_ = rng.randn(*state_shape).astype(dtype)
340 state_shape = sample_shape + independent_chain_shape + other_shape
341 state_ = rng.randn(*state_shape)
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
fusion_merger_test.cc 184 Shape state_shape = add_extra_input ? tuple_shape4_ : tuple_shape3_; local
187 HloInstruction::CreateParameter(0, state_shape, "state"));
  /external/tensorflow/tensorflow/python/ops/
rnn_cell_impl.py     [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 1813 const auto& state_shape = state_array.shape(); local
    [all...]

Completed in 347 milliseconds