| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 72 def lstm_model_fn(batch_size, state_size, steps, inputs_size=None): argument
85 def dynamic_lstm_model_fn(batch_size, state_size, max_steps): argument
111 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
119 def create_dynamic_lstm_batch_jacobian(batch_size, state_size, max_steps): argument
133 def create_lstm_batch_hessian(batch_size, state_size, steps): argument
145 def create_lstm_hessian(batch_size, state_size, steps): argument
185 def create_lstm_per_eg_grad(batch_size, state_size, steps, inputs_size=None): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 170 def _zero_state_tensors(state_size, batch_size, dtype): argument
266 def state_size(self): member in RNNCell
450 def state_size(self): member in BasicRNNCell
560 def state_size(self): member in GRUCell
737 def state_size(self): member in BasicLSTMCell
961 def state_size(self): member in LSTMCell
1277 def state_size(self): member in MultiRNNCell
|
| D | rnn_cell_wrapper_impl.py | 190 def state_size(self): member in DropoutWrapperBase
341 def state_size(self): member in ResidualWrapperBase
428 def state_size(self): member in DeviceWrapperBase
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops_gpu.cu.cc | 40 Distribution dist, int64 state_size, int64 output_size, in FillKernel()
80 auto state_size = state_tensor_flat.size(); in operator ()() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 60 def state_size(self): member in Plus1RNNCell
75 def state_size(self): member in ScalarStateRNNCell
93 def state_size(self): member in UnbalancedOutputRNNCell
112 def state_size(self): member in TensorArrayStateRNNCell
|
| D | rnn_cell_test.py | 67 def state_size(self): member in Plus1RNNCell
101 def state_size(self): member in DummyMultiDimensionalLSTM
120 def state_size(self): member in NestedRNNCell
131 def __init__(self, batch_size, state_size): argument
160 def state_size(self): member in TestStateSaver
174 def __init__(self, batch_size, state_size): argument
2373 def state_size(self): member in DeviceWrapperCell
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 91 def state_size(self): member in TfLiteRNNCell
289 def state_size(self): member in TFLiteLSTMCell
|
| /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/llvm-project/libcxx/test/std/numerics/rand/rand.eng/rand.eng.mers/ |
| D | ctor_sseq_all_zero.pass.cpp | 56 const std::size_t state_size = 1u; in test() local
|
| /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/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops.cc | 141 int state_size = counter_size + tensorflow::RNG_KEY_SIZE; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_test.py | 238 def state_size(self): member in RNNTest.test_minimal_rnn_cell_abstract_rnn_cell.MinimalRNNCell
895 def state_size(self): member in RNNTest.test_stacked_rnn_with_training_param.CellWrapper
1524 state_size = 1 variable in RNNTest.test_stateful_rnn_with_customized_get_initial_state.TestCell
|
| D | recurrent.py | 108 def state_size(self): member in StackedRNNCells
1081 def state_size(self): member in AbstractRNNCell
2992 def _is_multiple_state(state_size): argument
3005 def _generate_zero_filled_state(batch_size_tensor, state_size, dtype): argument
|
| D | wrappers_test.py | 99 def state_size(self): member in _AddOneCell
|
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| D | genX_blorp_exec.c | 132 unsigned state_size, unsigned state_alignment, in blorp_alloc_binding_table()
|
| /external/mesa3d/src/intel/vulkan/ |
| D | genX_blorp_exec.c | 110 unsigned state_size, unsigned state_alignment, in blorp_alloc_binding_table()
|
| /external/mesa3d/src/gallium/drivers/iris/ |
| D | iris_blorp.c | 145 unsigned state_size, in blorp_alloc_binding_table()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 366 int64 state_size; in RestoreInternal() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 158 state_size, argument
|
| D | stateful_random_ops.py | 100 def _make_1d_state(state_size, seed): argument
|
| /external/mesa3d/src/mesa/drivers/dri/radeon/ |
| D | radeon_tcl.c | 290 GLuint state_size; in radeonEnsureEmitSize() local
|
| /external/mesa3d/src/mesa/drivers/dri/r200/ |
| D | r200_tcl.c | 296 GLuint state_size; in r200EnsureEmitSize() local
|
| D | r200_state_init.c | 666 int state_size = TEX_STATE_SIZE_NEWDRM; in r200InitState() local
|
| /external/rust/crates/regex-automata/src/ |
| D | sparse.rs | 808 let state_size = size_of::<S>(); in to_bytes() localVariable
888 let state_size = NativeEndian::read_u16(buf) as usize; in from_bytes() localVariable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateful_random_ops.cc | 129 auto state_size = shape.dim_size(0); in CheckStateShape() local
|