Searched refs:state_size (Results 1 - 25 of 38) sorted by relevance

12

/external/mesa3d/src/gallium/drivers/ilo/core/
H A Dilo_builder_decode.c83 const unsigned state_size = sizeof(uint32_t); local
84 const unsigned count = item->size / state_size;
120 offset += state_size * 4;
129 const unsigned state_size = sizeof(uint32_t) * 4; local
130 const unsigned count = item->size / state_size;
149 offset += state_size;
158 const unsigned state_size = sizeof(uint32_t) * 16; local
159 const unsigned count = item->size / state_size;
210 offset += state_size;
219 const unsigned state_size local
265 const unsigned state_size = sizeof(uint32_t) * 2; local
292 const unsigned state_size = sizeof(uint32_t) * 2; local
370 const unsigned state_size = sizeof(uint32_t) * 2; local
393 const unsigned state_size = sizeof(uint32_t) * 4; local
420 const unsigned state_size = sizeof(uint32_t) * 8; local
569 const unsigned state_size = sizeof(uint32_t) * 1; local
[all...]
/external/libcxx/test/std/numerics/rand/rand.eng/rand.eng.mers/
H A Dvalues.pass.cpp23 // static constexpr size_t state_size = n;
53 static_assert((E::state_size == 624), "");
74 where(E::state_size);
94 static_assert((E::state_size == 312), "");
115 where(E::state_size);
/external/tensorflow/tensorflow/python/ops/
H A Drnn_cell_impl.py59 conditions = [hasattr(cell, "output_size"), hasattr(cell, "state_size"),
124 def _zero_state_tensors(state_size, batch_size, dtype):
125 """Create tensors of zeros based on state_size, batch_size, and dtype."""
134 return nest.map_structure(get_state_shape, state_size)
153 If `self.state_size` is an integer, this operation also results in a new
154 state matrix with `self.state_size` columns. If `self.state_size` is a
165 state: if `self.state_size` is an integer, this should be a `2-D Tensor`
166 with shape `[batch_size, self.state_size]`. Otherwise, if
167 `self.state_size` i
207 def state_size(self): member in class:RNNCell
324 def state_size(self): member in class:BasicRNNCell
393 def state_size(self): member in class:GRUCell
528 def state_size(self): member in class:BasicLSTMCell
702 def state_size(self): member in class:LSTMCell
995 def state_size(self): member in class:DropoutWrapper
1091 def state_size(self): member in class:ResidualWrapper
1146 def state_size(self): member in class:DeviceWrapper
1198 def state_size(self): member in class:MultiRNNCell
1275 def state_size(self): member in class:_SlimRNNCell
[all...]
H A Drnn.py150 zero_output, state, call_cell, state_size, skip_conditionals=False):
180 state: Either a single `Tensor` matrix of shape `[batch_size, state_size]`,
184 new_state is a `Tensor` matrix of shape `[batch_size, state_size]`.
185 state_size: The `cell.state_size` associated with the state.
199 that returned by `state_size`.
355 `[batch_size, cell_fw.state_size]`.
356 If `cell_fw.state_size` is a tuple, this should be a tuple of
357 tensors having shapes `[batch_size, s] for s in cell_fw.state_size`.
519 If `cell.state_size` i
[all...]
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
H A Dcore_rnn_cell.py235 def state_size(self): member in class:EmbeddingWrapper
236 return self._cell.state_size
312 def state_size(self): member in class:InputProjectionWrapper
313 return self._cell.state_size
369 def state_size(self): member in class:OutputProjectionWrapper
370 return self._cell.state_size
H A Drnn_cell.py208 def state_size(self): member in class:CoupledInputForgetGateLSTMCell
221 `2-D, batch x state_size`. If `state_is_tuple` is True, this must be a
379 def state_size(self): member in class:TimeFreqLSTMCell
387 state: state Tensor, 2D, batch x state_size.
394 - A 2D, batch x state_size, Tensor representing the new state of LSTM
609 def state_size(self): member in class:GridLSTMCell
621 state: Tensor or tuple of Tensors, 2D, [batch, state_size], depends on the
629 - A 2D, [batch, state_size], Tensor representing the new state of LSTM
667 state: Tensor or tuple of Tensors, 2D, [batch, state_size], it depends on
678 - A list of [batch, state_size] Tensor
1174 def state_size(self): member in class:AttentionCellWrapper
1277 def state_size(self): member in class:HighwayWrapper
1401 def state_size(self): member in class:LayerNormBasicLSTMCell
1498 def state_size(self): member in class:NASCell
1647 def state_size(self): member in class:UGRNNCell
1756 def state_size(self): member in class:IntersectionRNNCell
1844 def state_size(self): member in class:CompiledWrapper
1945 def state_size(self): member in class:PhasedLSTMCell
2105 def state_size(self): member in class:ConvLSTMCell
2293 def state_size(self): member in class:GLSTMCell
2514 def state_size(self): member in class:LayerNormLSTMCell
2721 def state_size(self): member in class:SRUCell
2846 def state_size(self): member in class:WeightNormLSTMCell
[all...]
H A Dgru_ops.py170 def state_size(self): member in class:GRUBlockCell
198 # Check cell_size == state_size from h_prev.
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
H A Dstate_saving_rnn_estimator.py56 `cell.state_size` is a tuple) then `state_name` should be a tuple of
57 strings having the same length as `cell.state_size`. Otherwise it should
195 string pairs, depending on the type of `cell.state_size`.
198 TypeError: If cell.state_size is of type TensorShape.
200 state_size = cell.state_size
201 if isinstance(state_size, tensor_shape.TensorShape):
202 raise TypeError('cell.state_size of type TensorShape is not supported.')
203 if isinstance(state_size, int):
205 if isinstance(state_size, rnn_cel
[all...]
H A Ddynamic_rnn_estimator.py81 where `n` is the number of nested entries in `cell.state_size`, this
82 function returns `None`. Otherwise, returns a `Tensor` if `cell.state_size`
83 is an `int` or a nested tuple of `Tensor`s if `cell.state_size` is a nested
89 flat_state_sizes = nest.flatten(cell.state_size)
92 for i, state_size in enumerate(flat_state_sizes):
100 state_size,
574 elements in `cell.state_size`. The input function must contain values for
583 in `cell.state_size`, along with `PredictionKey.CLASSES` for problem type
H A Ddynamic_rnn_estimator_test.py51 def __init__(self, state_size, output_size):
52 self._state_size = state_size
56 def state_size(self): member in class:IdentityRNNCell
65 [array_ops.shape(inputs)[0], self.state_size])
409 for i, state_size in enumerate([4, 4, 8, 8, 7, 7]):
411 self.assertListEqual(list(state_piece.shape), [batch_size, state_size])
/external/tensorflow/tensorflow/python/keras/_impl/keras/layers/
H A Drecurrent.py67 if not hasattr(cell, 'state_size'):
69 '`state_size` attribute. '
75 def state_size(self): member in class:StackedRNNCells
79 # `stack.state_size[0] == output_dim`.
83 state_size = []
85 if hasattr(cell.state_size, '__len__'):
86 state_size += list(cell.state_size)
88 state_size.append(cell.state_size)
[all...]
H A Drecurrent_test.py38 self.state_size = units
73 self.state_size = (units, units)
100 assert layer.cell.state_size == (32, 32, 16, 16, 8, 8)
112 self.state_size = units
188 self.state_size = units
300 self.state_size = units
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
H A Dcore_rnn_cell_test.py242 state_size = num_units * 2
246 m = array_ops.zeros([batch_size - 1, state_size])
254 m.name: 0.1 * np.ones([batch_size - 1, state_size])
258 """Tests that state_size must be num_units * 2."""
263 state_size = num_units * 3 # state_size must be num_units * 2
267 m = array_ops.zeros([batch_size, state_size])
275 m.name: 0.1 * np.ones([batch_size, state_size])
288 self.assertTrue(isinstance(cell.state_size, tuple))
290 isinstance(cell.state_size[
[all...]
H A Dcore_rnn_test.py57 def state_size(self): member in class:Plus1RNNCell
91 def state_size(self): member in class:DummyMultiDimensionalLSTM
110 def state_size(self): member in class:NestedRNNCell
121 def __init__(self, batch_size, state_size):
123 self._state_size = state_size
129 state_size = self._state_size[name]
131 state_size = self._state_size
132 if isinstance(state_size, int):
133 state_size = (state_size,)
2227 def state_size(self): member in class:DeviceWrapperCell
[all...]
/external/tensorflow/tensorflow/contrib/grid_rnn/python/kernel_tests/
H A Dgrid_rnn_test.py44 self.assertEqual(cell.state_size, ((2, 2), (2, 2)))
102 self.assertEqual(cell.state_size, ((2, 2), (2, 2)))
145 self.assertEqual(cell.state_size, ((2, 2),))
173 self.assertEqual(cell.state_size, ((2, 2), (2, 2)))
208 self.assertEqual(cell.state_size, ((2, 2), (2, 2)))
243 self.assertEqual(cell.state_size, ((2, 2),))
270 self.assertEqual(cell.state_size, (2, 2))
301 self.assertEqual(cell.state_size, (2, 2))
332 self.assertEqual(cell.state_size, (2,))
357 self.assertEqual(cell.state_size, ((
[all...]
/external/mesa3d/src/mesa/drivers/dri/radeon/
H A Dradeon_tcl.c289 GLuint state_size; local
314 state_size = radeonCountStateEmitSize( &rmesa->radeon );
317 state_size += rmesa->hw.tcl.check( &rmesa->radeon.glCtx, &rmesa->hw.tcl );
344 return space_required + state_size;
/external/tensorflow/tensorflow/core/profiler/g3doc/
H A Dprofile_time.md92 core_rnn.py:195:static_rnn:state_size=cell.s... (0us/127.20ms)
94 core_rnn.py:195:static_rnn:state_size=cell.s... (0us/125.86ms)
H A Dadvise.md59 core_rnn.py:195:static_rnn:state_size=cell.s..., cpu: 306.52ms, accelerator: 73.54ms, total: 380.05ms
64 core_rnn.py:195:static_rnn:state_size=cell.s..., cpu: 296.11ms, accelerator: 73.54ms, total: 369.65ms
/external/mesa3d/src/mesa/drivers/dri/i965/
H A DgenX_blorp_exec.c107 unsigned state_size, unsigned state_alignment,
120 state_size, state_alignment,
106 blorp_alloc_binding_table(struct blorp_batch *batch, unsigned num_entries, unsigned state_size, unsigned state_alignment, uint32_t *bt_offset, uint32_t *surface_offsets, void **surface_maps) argument
/external/mesa3d/src/mesa/drivers/dri/r200/
H A Dr200_tcl.c296 GLuint state_size; local
311 state_size = radeonCountStateEmitSize( &rmesa->radeon );
314 state_size += rmesa->hw.vtx.check(&rmesa->radeon.glCtx, &rmesa->hw.vtx);
341 if (rcommonEnsureCmdBufSpace(&rmesa->radeon, space_required + state_size, __func__))
344 return space_required + state_size;
/external/mesa3d/src/intel/vulkan/
H A Danv_allocator.c590 size_t state_size)
593 assert(state_size >= 64 && util_is_power_of_two(state_size));
595 pool->state_size = state_size;
617 block.u64 = __sync_fetch_and_add(&pool->block.u64, pool->state_size);
623 new.next = offset + pool->state_size;
589 anv_fixed_size_state_pool_init(struct anv_fixed_size_state_pool *pool, size_t state_size) argument
H A DgenX_blorp_exec.c82 unsigned state_size, unsigned state_alignment,
81 blorp_alloc_binding_table(struct blorp_batch *batch, unsigned num_entries, unsigned state_size, unsigned state_alignment, uint32_t *bt_offset, uint32_t *surface_offsets, void **surface_maps) argument
/external/tensorflow/tensorflow/contrib/crf/python/ops/
H A Dcrf.py280 def state_size(self): member in class:CrfForwardRnnCell
366 def state_size(self): member in class:CrfDecodeForwardRnnCell
413 def state_size(self): member in class:CrfDecodeBackwardRnnCell
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
H A Dgrid_rnn_cell.py147 self._state_size = tuple(self._cells[0].state_size
157 def state_size(self): member in class:GridRNNCell
165 state: state Tensor, 2D, batch x state_size. Note that state_size =
174 - A 2D, batch x state_size, Tensor representing the new state of the cell
258 range(0, self.state_size,
314 state_sizes = self._cells[0].state_size
/external/v8/src/compiler/
H A Dmemory-optimizer.cc171 int32_t const state_size = state->size() + object_size; local
175 if (OpParameter<int32_t>(group->size()) < state_size) {
177 common()->Int32Constant(state_size));
193 state = AllocationState::Open(group, state_size, top, zone());

Completed in 2061 milliseconds

12