Home
last modified time | relevance | path

Searched defs:initial_state (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc141 XlaOp initial_state, const Shape& shape) { in GetThreeFryInputsAndUpdatedState()
244 RngOutput ThreeFryRngBit32(XlaOp key, XlaOp initial_state, const Shape& shape) { in ThreeFryRngBit32()
256 RngOutput ThreeFryRngBit64(XlaOp key, XlaOp initial_state, const Shape& shape) { in ThreeFryRngBit64()
392 XlaOp initial_state, in GeneratePhiloxBits()
408 RngOutput PhiloxRngBit32(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit32()
435 RngOutput PhiloxRngBit64(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit64()
533 RngOutput ThreeFryBitGenerator(XlaOp key, XlaOp initial_state, in ThreeFryBitGenerator()
554 RngOutput PhiloxBitGenerator(XlaOp key, XlaOp initial_state, in PhiloxBitGenerator()
582 RngOutput UniformFloatingPointDistribution(XlaOp key, XlaOp initial_state, in UniformFloatingPointDistribution()
593 RngOutput UniformIntDistribution(XlaOp key, XlaOp initial_state, in UniformIntDistribution()
[all …]
/external/mesa3d/src/intel/isl/
Disl_aux_info.c130 isl_aux_prepare_access(enum isl_aux_state 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()
208 isl_aux_state_transition_write(enum isl_aux_state initial_state, in isl_aux_state_transition_write()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dscan_ops.py26 def scan(initial_state, scan_func): argument
/external/tensorflow/tensorflow/python/keras/layers/
Dcudnn_recurrent.py79 def call(self, inputs, mask=None, training=None, initial_state=None): argument
268 def _process_batch(self, inputs, initial_state): argument
464 def _process_batch(self, inputs, initial_state): argument
Dwrappers.py577 def __call__(self, inputs, initial_state=None, constants=None, **kwargs): argument
656 initial_state=None, argument
Drecurrent.py661 def __call__(self, inputs, initial_state=None, constants=None, **kwargs): argument
731 initial_state=None, argument
836 def _process_inputs(self, inputs, initial_state, constants): argument
1586 def call(self, inputs, mask=None, training=None, initial_state=None): argument
2128 def call(self, inputs, mask=None, training=None, initial_state=None): argument
2801 def call(self, inputs, mask=None, training=None, initial_state=None): argument
2936 def _standardize_args(inputs, initial_state, constants, num_constants): argument
Drecurrent_v2.py418 def call(self, inputs, mask=None, training=None, initial_state=None): argument
477 def _defun_gru_call(self, inputs, initial_state, training, mask, argument
1145 def call(self, inputs, mask=None, training=None, initial_state=None): argument
Dconvolutional_recurrent.py295 initial_state=None, argument
903 def call(self, inputs, mask=None, training=None, initial_state=None): argument
/external/rust/crates/grpcio-sys/grpc/src/core/ext/filters/client_channel/
Dsubchannel_interface.h109 grpc_connectivity_state initial_state, in WatchConnectivityState()
Dsubchannel.cc446 grpc_connectivity_state initial_state, in AddWatcherLocked()
524 grpc_connectivity_state initial_state, in AddWatcherLocked()
783 grpc_connectivity_state initial_state, in WatchConnectivityState()
/external/tensorflow/tensorflow/python/types/
Ddistribute.py32 def reduce(self, initial_state, reduce_func): argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateless_random_ops.cc86 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniform() local
122 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniformFullInt() local
305 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in Compile() local
Dstateful_random_ops.cc67 xla::XlaOp initial_state, in StatefulRngUniform()
93 xla::XlaOp initial_state, in StatefulRngUniformFullInt()
/external/yapf/yapf/yapflib/
Dreformatter.py308 def _AnalyzeSolutionSpace(initial_state): argument
393 def _ReconstructPath(initial_state, current): argument
/external/webrtc/modules/audio_processing/aec3/
Dsubband_nearend_detector.cc36 bool initial_state) { in Update()
Ddominant_nearend_detector.cc36 bool initial_state) { in Update()
/external/tensorflow/tensorflow/python/ops/
Drnn.py508 initial_state=None, argument
766 initial_state, argument
1309 initial_state=None, argument
/external/libchrome/base/synchronization/
Dwaitable_event_posix.cc46 InitialState initial_state) in WaitableEvent()
373 InitialState initial_state) in WaitableEventKernel()
/external/libchrome/base/threading/
Dthread_perftest.cc236 WaitableEvent::InitialState initial_state) in ConditionVariableEvent()
276 WaitableEvent::InitialState initial_state) { in PthreadEvent()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dscan_dataset_op.cc56 std::vector<Tensor> initial_state(initial_state_inputs.begin(), in MakeDataset() local
74 std::vector<Tensor> initial_state, in Dataset()
/external/rust/crates/grpcio-sys/grpc/src/core/lib/transport/
Dconnectivity_state.cc118 grpc_connectivity_state initial_state, in AddWatcher()
/external/tensorflow/tensorflow/core/kernels/data/
Dreduce_dataset_op_test.cc27 ReduceDatasetParams(T input_dataset_params, std::vector<Tensor> initial_state, in ReduceDatasetParams()
/external/webrtc/modules/congestion_controller/goog_cc/
Dtrendline_estimator_unittest.cc59 auto initial_state = estimator.State(); in RunTestUntilStateChange() local
/external/tensorflow/tensorflow/lite/kernels/
Dquant_basic_lstm_test.cc123 std::vector<int16_t> initial_state(GetTensorSize(cell_state_out_), 0); in QuantizedLSTMOpModel() local
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc821 shape_inference::ShapeHandle initial_state; in __anon592c2f360b02() local

12