| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 68 def lstm_model_fn(batch_size, state_size, steps, inputs_size=None): argument
81 def dynamic_lstm_model_fn(batch_size, state_size, max_steps): argument
107 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
115 def create_dynamic_lstm_batch_jacobian(batch_size, state_size, max_steps): argument
129 def create_lstm_batch_hessian(batch_size, state_size, steps): argument
141 def create_lstm_hessian(batch_size, state_size, steps): argument
181 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 | 168 def _zero_state_tensors(state_size, batch_size, dtype): argument
265 def state_size(self): member in RNNCell
450 def state_size(self): member in BasicRNNCell
561 def state_size(self): member in GRUCell
740 def state_size(self): member in BasicLSTMCell
965 def state_size(self): member in LSTMCell
1285 def state_size(self): member in MultiRNNCell
|
| D | rnn_cell_wrapper_impl.py | 186 def state_size(self): member in DropoutWrapperBase
337 def state_size(self): member in ResidualWrapperBase
424 def state_size(self): member in DeviceWrapperBase
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_reduce_dataset.cc | 147 const int state_size = in CreateOptionalDatasetIf() local
249 const int state_size = in PopulateDatasetWhileBody() local
292 const int state_size = in DecomposeReduceDatasetInFunction() local
|
| D | decompose_resource_ops.cc | 128 int state_size = counter_size + tensorflow::RNG_KEY_SIZE; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | rnn_test.py | 55 def state_size(self): member in Plus1RNNCell
70 def state_size(self): member in ScalarStateRNNCell
88 def state_size(self): member in UnbalancedOutputRNNCell
107 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
2451 def state_size(self): member in DeviceWrapperCell
|
| /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/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/cronet/buildtools/third_party/libc++/trunk/test/std/numerics/rand/rand.eng/rand.eng.mers/ |
| D | ctor_sseq_all_zero.pass.cpp | 53 const std::size_t state_size = 1u; in test() local
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | session_debug_testlib.py | 65 def __init__(self, input_output_size, state_size): argument
75 def state_size(self): member in _RNNCellForTest
|
| /external/cronet/buildtools/third_party/libc++/trunk/include/__random/ |
| D | mersenne_twister_engine.h | 108 static _LIBCPP_CONSTEXPR const size_t state_size = __n; variable
|
| /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/mesa/drivers/dri/i965/ |
| D | genX_blorp_exec.c | 132 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/python/keras/layers/ |
| D | recurrent.py | 104 def state_size(self): member in StackedRNNCells
1079 def state_size(self): member in AbstractRNNCell
2997 def _is_multiple_state(state_size): argument
3010 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 | 366 int64_t state_size; in RestoreInternal() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 155 state_size, argument
|
| D | stateful_random_ops.py | 91 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
|
| D | r200_swtcl.c | 208 const int state_size = radeonCountStateEmitSize(&rmesa->radeon); in r200_predict_emit_size() 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 | 130 auto state_size = shape.dim_size(0); in CheckStateShape() local
|