| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops_gpu.cu.cc | 32 Distribution dist, int64 state_size, int64 output_size, in FillKernel()
66 auto state_size = state_tensor_flat.size(); in operator ()() local
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 73 def lstm_model_fn(batch_size, state_size, steps, inputs_size=None): argument
86 def dynamic_lstm_model_fn(batch_size, state_size, max_steps): argument
112 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
120 def create_dynamic_lstm_batch_jacobian(batch_size, state_size, max_steps): argument
134 def create_lstm_batch_hessian(batch_size, state_size, steps): argument
146 def create_lstm_hessian(batch_size, state_size, steps): argument
186 def create_lstm_per_eg_grad(batch_size, state_size, steps, inputs_size=None): argument
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 213 def state_size(self): member in CoupledInputForgetGateLSTMCell
386 def state_size(self): member in TimeFreqLSTMCell
616 def state_size(self): member in GridLSTMCell
1182 def state_size(self): member in AttentionCellWrapper
1285 def state_size(self): member in HighwayWrapper
1409 def state_size(self): member in LayerNormBasicLSTMCell
1511 def state_size(self): member in NASCell
1667 def state_size(self): member in UGRNNCell
1776 def state_size(self): member in IntersectionRNNCell
1864 def state_size(self): member in CompiledWrapper
[all …]
|
| D | core_rnn_cell.py | 233 def state_size(self): member in EmbeddingWrapper
309 def state_size(self): member in InputProjectionWrapper
365 def state_size(self): member in OutputProjectionWrapper
|
| D | gru_ops.py | 171 def state_size(self): member in GRUBlockCell
|
| D | lstm_ops.py | 392 def state_size(self): member in LSTMBlockCell
|
| /external/tensorflow/tensorflow/contrib/crf/python/ops/ |
| D | crf.py | 355 def state_size(self): member in CrfForwardRnnCell
441 def state_size(self): member in CrfDecodeForwardRnnCell
488 def state_size(self): member in CrfDecodeBackwardRnnCell
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 171 def _zero_state_tensors(state_size, batch_size, dtype): argument
264 def state_size(self): member in RNNCell
436 def state_size(self): member in BasicRNNCell
537 def state_size(self): member in GRUCell
715 def state_size(self): member in BasicLSTMCell
924 def state_size(self): member in LSTMCell
1340 def state_size(self): member in DropoutWrapperBase
1469 def state_size(self): member in ResidualWrapperBase
1545 def state_size(self): member in DeviceWrapperBase
1639 def state_size(self): member in MultiRNNCell
|
| D | stateful_random_ops.py | 87 def _make_1d_state(state_size, seed): argument
|
| D | rnn.py | 174 zero_output, state, call_cell, state_size, skip_conditionals=False): argument
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 87 def state_size(self): member in TfLiteRNNCell
277 def state_size(self): member in TFLiteLSTMCell
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 67 def state_size(self): member in Plus1RNNCell
82 def state_size(self): member in ScalarStateRNNCell
100 def state_size(self): member in UnbalancedOutputRNNCell
119 def state_size(self): member in TensorArrayStateRNNCell
|
| D | rnn_cell_test.py | 64 def state_size(self): member in Plus1RNNCell
98 def state_size(self): member in DummyMultiDimensionalLSTM
117 def state_size(self): member in NestedRNNCell
128 def __init__(self, batch_size, state_size): argument
157 def state_size(self): member in TestStateSaver
171 def __init__(self, batch_size, state_size): argument
2363 def state_size(self): member in DeviceWrapperCell
|
| /external/libcxx/test/std/numerics/rand/rand.eng/rand.eng.mers/ |
| D | ctor_sseq_all_zero.pass.cpp | 55 const std::size_t state_size = 1u; in test() local
|
| /external/tensorflow/tensorflow/examples/saved_model/integration_tests/ |
| D | export_text_rnn_model.py | 38 def __init__(self, vocab, emb_dim, buckets, state_size): argument
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | session_debug_testlib.py | 70 def __init__(self, input_output_size, state_size): argument
80 def state_size(self): member in _RNNCellForTest
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dynamic_rnn_estimator_test.py | 51 def __init__(self, state_size, output_size): argument
56 def state_size(self): member in IdentityRNNCell
|
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| D | genX_blorp_exec.c | 124 unsigned state_size, unsigned state_alignment, in blorp_alloc_binding_table()
|
| /external/mesa3d/src/intel/vulkan/ |
| D | genX_blorp_exec.c | 94 unsigned state_size, unsigned state_alignment, in blorp_alloc_binding_table()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | attention_wrapper.py | 75 def state_size(self): member in AttentionMechanism
184 def state_size(self): member in _BaseAttentionMechanism
554 def state_size(self): member in _BaseAttentionMechanismV2
2308 def state_size(self): member in AttentionWrapper
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent.py | 91 def state_size(self): member in StackedRNNCells
983 def state_size(self): member in AbstractRNNCell
2742 def _is_multiple_state(state_size): argument
2755 def _generate_zero_filled_state(batch_size_tensor, state_size, dtype): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 344 int64 state_size; in RestoreInternal() local
|
| /external/mesa3d/src/mesa/drivers/dri/radeon/ |
| D | radeon_tcl.c | 291 GLuint state_size; in radeonEnsureEmitSize() local
|
| /external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
| D | grid_rnn_cell.py | 158 def state_size(self): member in GridRNNCell
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateful_random_ops.cc | 206 auto state_size = var_shape.dim_size(0); in CompileImpl() local
|