Home
last modified time | relevance | path

Searched refs:initial_state (Results 1 – 25 of 165) sorted by relevance

1234567

/external/mesa3d/src/intel/isl/
Disl_aux_info.c130 isl_aux_prepare_access(enum isl_aux_state initial_state, in isl_aux_prepare_access() argument
137 assert(aux_state_possible(initial_state, state_superset_usage)); in isl_aux_prepare_access()
141 switch (initial_state) { in isl_aux_prepare_access()
171 isl_aux_state_transition_aux_op(enum isl_aux_state initial_state, in isl_aux_state_transition_aux_op() argument
175 assert(aux_state_possible(initial_state, usage)); in isl_aux_state_transition_aux_op()
180 return initial_state; in isl_aux_state_transition_aux_op()
185 assert(isl_aux_state_has_valid_aux(initial_state)); in isl_aux_state_transition_aux_op()
187 return initial_state == ISL_AUX_STATE_CLEAR || in isl_aux_state_transition_aux_op()
188 initial_state == ISL_AUX_STATE_PARTIAL_CLEAR || in isl_aux_state_transition_aux_op()
189 initial_state == ISL_AUX_STATE_COMPRESSED_CLEAR ? in isl_aux_state_transition_aux_op()
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dlstm_test.py198 initial_state = [keras.Input((units,)) for _ in range(num_states)]
200 if len(initial_state) == 1:
201 output = layer(inputs, initial_state=initial_state[0])
203 output = layer(inputs, initial_state=initial_state)
205 any(initial_state[0] is t
208 model = keras.models.Model([inputs] + initial_state, output)
215 initial_state = [np.random.random((num_samples, units))
218 model.train_on_batch([inputs] + initial_state, targets)
229 initial_state = [keras.backend.random_normal_variable(
233 output = layer(inputs, initial_state=initial_state)
[all …]
Dwrappers.py577 def __call__(self, inputs, initial_state=None, constants=None, **kwargs): argument
579 inputs, initial_state, constants = _standardize_args(
580 inputs, initial_state, constants, self._num_constants)
584 initial_state = inputs[1:]
587 if initial_state is None and constants is None:
593 if initial_state is not None:
595 num_states = len(initial_state)
601 'Found: ' + str(initial_state))
603 kwargs['initial_state'] = initial_state
604 additional_inputs += initial_state
[all …]
Dcudnn_recurrent.py79 def call(self, inputs, mask=None, training=None, initial_state=None): argument
89 initial_state = inputs[1:]
91 elif initial_state is not None:
94 initial_state = self.states
96 initial_state = self.get_initial_state(inputs)
98 if len(initial_state) != len(self.states):
100 ' states but was passed ' + str(len(initial_state)) +
106 output, states = self._process_batch(inputs, initial_state)
268 def _process_batch(self, inputs, initial_state): argument
271 input_h = initial_state[0]
[all …]
Dconvolutional_recurrent.py276 initial_state = backend.zeros_like(inputs)
278 initial_state = backend.sum(initial_state, axis=1)
281 initial_state = self.cell.input_conv(initial_state,
283 initial_state.dtype),
287 return [initial_state for _ in self.cell.state_size]
289 return [initial_state]
295 initial_state=None, argument
299 inputs, initial_state, constants = self._process_inputs(
300 inputs, initial_state, constants)
324 initial_state,
[all …]
Drecurrent.py661 def __call__(self, inputs, initial_state=None, constants=None, **kwargs): argument
662 inputs, initial_state, constants = _standardize_args(inputs,
663 initial_state,
667 if initial_state is None and constants is None:
676 if initial_state is not None:
677 additional_inputs += initial_state
679 lambda s: InputSpec(shape=backend.int_shape(s)), initial_state)
721 if initial_state is not None:
722 kwargs['initial_state'] = initial_state
731 initial_state=None, argument
[all …]
Dlstm_v2_test.py144 initial_state = [keras.Input((units,)) for _ in range(num_states)]
146 if len(initial_state) == 1:
147 output = layer(inputs, initial_state=initial_state[0])
149 output = layer(inputs, initial_state=initial_state)
151 any(initial_state[0] is t
154 model = keras.models.Model([inputs] + initial_state, output)
160 initial_state = [
164 model.train_on_batch([inputs] + initial_state, targets)
175 initial_state = [
180 output = layer(inputs, initial_state=initial_state)
[all …]
Dcudnn_recurrent_test.py134 initial_state = [keras.Input((units,)) for _ in range(num_states)]
136 if len(initial_state) == 1:
137 output = layer(inputs, initial_state=initial_state[0])
139 output = layer(inputs, initial_state=initial_state)
141 any(initial_state[0] is t
144 model = keras.models.Model([inputs] + initial_state, output)
151 initial_state = [
155 model.fit([inputs] + initial_state, targets)
Drecurrent_v2.py418 def call(self, inputs, mask=None, training=None, initial_state=None): argument
426 inputs, initial_state, _ = self._process_inputs(inputs, initial_state, None)
446 initial_state,
458 inputs, initial_state, training, mask, row_lengths)
477 def _defun_gru_call(self, inputs, initial_state, training, mask, argument
491 'init_h': _read_variable_value(initial_state[0]),
506 'init_h': _read_variable_value(initial_state[0]),
1145 def call(self, inputs, mask=None, training=None, initial_state=None): argument
1153 inputs, initial_state, _ = self._process_inputs(inputs, initial_state, None)
1174 initial_state,
[all …]
Drecurrent_test.py502 y = layer(x, initial_state=s, constants=c)
523 y = layer(x, initial_state=s, constants=c)
550 y = layer(x, initial_state=s, constants=c)
567 y = layer(x, initial_state=s, constants=c)
661 output = layer_class(units)(input2, initial_state=state)
998 y = layer(x, initial_state=s)
1020 layer(x, initial_state=s)
1056 initial_state = cell.get_initial_state(inputs, None, None)
1057 self.assertEqual(initial_state.shape.as_list(), [None, 5])
1058 self.assertEqual(initial_state.dtype, inputs.dtype)
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.h44 using BitGeneratorTy = std::function<RngOutput(XlaOp key, XlaOp initial_state,
50 RngOutput ThreeFryBitGenerator(XlaOp key, XlaOp initial_state,
61 RngOutput PhiloxBitGenerator(XlaOp key, XlaOp initial_state,
70 RngOutput UniformFloatingPointDistribution(XlaOp key, XlaOp initial_state,
77 RngOutput UniformIntDistribution(XlaOp key, XlaOp initial_state,
84 RngOutput NormalFloatingPointDistribution(XlaOp key, XlaOp initial_state,
Dprng.cc141 XlaOp initial_state, const Shape& shape) { in GetThreeFryInputsAndUpdatedState() argument
142 XlaBuilder* builder = initial_state.builder(); in GetThreeFryInputsAndUpdatedState()
145 auto input_u64 = Broadcast(Reshape(initial_state, {}), shape.dimensions()); in GetThreeFryInputsAndUpdatedState()
157 initial_state + ConstantR0<uint64>(builder, ShapeUtil::ElementsIn(shape)); in GetThreeFryInputsAndUpdatedState()
244 RngOutput ThreeFryRngBit32(XlaOp key, XlaOp initial_state, const Shape& shape) { in ThreeFryRngBit32() argument
247 GetThreeFryInputsAndUpdatedState(initial_state, shape_pair.half_shape); in ThreeFryRngBit32()
256 RngOutput ThreeFryRngBit64(XlaOp key, XlaOp initial_state, const Shape& shape) { in ThreeFryRngBit64() argument
258 GetThreeFryInputsAndUpdatedState(initial_state, shape); in ThreeFryRngBit64()
392 XlaOp initial_state, in GeneratePhiloxBits() argument
395 state = Uint128ToUint32s(Uint128FromOp(initial_state)); in GeneratePhiloxBits()
[all …]
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dscan_ops.py26 def scan(initial_state, scan_func): argument
47 return dataset.scan(initial_state=initial_state, scan_func=scan_func)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dscan_test.py50 initial_state=start, scan_func=scan_fn)
74 initial_state=[0, 1],
122 initial_state=start, scan_func=scan_fn)
153 initial_state=start, scan_func=scan_fn)
191 dataset_ops.Dataset.range(6).scan(initial_state=start, scan_func=scan_fn)
206 initial_state=([0], 1), scan_func=_scan_fn)
235 initial_state=constant_op.constant(1, dtype=dtypes.int32),
250 initial_state=constant_op.constant(1, dtype=dtypes.int32),
264 initial_state=constant_op.constant(1), scan_func=scan_fn)
304 initial_state=[0, 1],
/external/tensorflow/tensorflow/python/ops/
Drnn.py442 initial_state=initial_state_fw,
481 initial_state=initial_state_bw,
508 initial_state=None, argument
716 if initial_state is not None:
717 state = initial_state
766 initial_state, argument
805 state = initial_state
1162 initial_state, emit_structure, init_loop_state) = loop_fn(
1184 nest.assert_same_structure(initial_state, cell.state_size)
1185 state = initial_state
[all …]
/external/rust/crates/ring/src/
Ddigest.rs55 state: algorithm.initial_state, in new()
284 initial_state: State, field
318 initial_state: State {
343 initial_state: State {
368 initial_state: State {
393 initial_state: State {
422 initial_state: State {
579 state: alg.initial_state, in nearly_full_context()
/external/rust/crates/grpcio-sys/grpc/src/core/ext/filters/client_channel/
Dsubchannel_interface.h70 grpc_connectivity_state initial_state,
109 grpc_connectivity_state initial_state, in WatchConnectivityState() argument
111 return wrapped_subchannel_->WatchConnectivityState(initial_state, in WatchConnectivityState()
/external/elfutils/libdw/
Dcfi.c316 cfi_assert (cie->initial_state != NULL); in execute_cfi()
321 if (cie->initial_state->nregs > operand) in execute_cfi()
322 fs->regs[operand] = cie->initial_state->regs[operand]; in execute_cfi()
396 cfi_assert (cie->initial_state != NULL); in execute_cfi()
444 if (likely (cie->initial_state != NULL)) in cie_cache_initial_state()
501 cie->initial_state = cie_fs; in cie_cache_initial_state()
515 Dwarf_Frame *fs = duplicate_frame_state (fde->cie->initial_state, NULL); in __libdw_frame_at_address()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateless_random_ops.cc86 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniform() local
92 key, initial_state, in StatelessRngUniform()
99 key, initial_state, in StatelessRngUniform()
122 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniformFullInt() local
125 GetBitGeneratorForDevice(device_type_string)(key, initial_state, shape); in StatelessRngUniformFullInt()
305 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in Compile() local
310 key, initial_state, GetBitGeneratorForDevice(device_type_string_), in Compile()
Dstateful_random_ops.cc67 xla::XlaOp initial_state, in StatefulRngUniform() argument
75 key, initial_state, BitGen(alg), minval, maxval, shape); in StatefulRngUniform()
80 return UniformIntDistribution(key, initial_state, BitGen(alg), minval, in StatefulRngUniform()
88 initial_state}; in StatefulRngUniform()
93 xla::XlaOp initial_state, in StatefulRngUniformFullInt() argument
96 xla::RngOutput output = BitGen(alg)(key, initial_state, shape); in StatefulRngUniformFullInt()
111 initial_state}; in StatefulRngUniformFullInt()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ReduceDataset.pbtxt11 name: "initial_state"
22 `new_state` must match the structure of `initial_state`.
/external/rust/crates/grpcio-sys/grpc/src/core/lib/transport/
Dconnectivity_state.cc118 grpc_connectivity_state initial_state, in AddWatcher() argument
125 if (initial_state != current_state) { in AddWatcher()
129 name_, this, watcher.get(), ConnectivityStateName(initial_state), in AddWatcher()
/external/libchrome/base/threading/
Dthread_perftest.cc236 WaitableEvent::InitialState initial_state) in ConditionVariableEvent() argument
239 DCHECK_EQ(WaitableEvent::InitialState::NOT_SIGNALED, initial_state); in ConditionVariableEvent()
276 WaitableEvent::InitialState initial_state) { in PthreadEvent() argument
278 DCHECK_EQ(WaitableEvent::InitialState::NOT_SIGNALED, initial_state); in PthreadEvent()
/external/webrtc/modules/audio_processing/aec3/
Ddominant_nearend_detector.cc36 bool initial_state) { in Update() argument
51 if ((!initial_state || use_during_initial_phase_) && in Update()
/external/libchrome/base/synchronization/
Dwaitable_event_posix.cc46 InitialState initial_state) in WaitableEvent() argument
47 : kernel_(new WaitableEventKernel(reset_policy, initial_state)) {} in WaitableEvent()
373 InitialState initial_state) in WaitableEventKernel() argument
375 signaled_(initial_state == InitialState::SIGNALED) {} in WaitableEventKernel()

1234567