Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dprng.cc139 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 …]
/aosp_15_r20/external/mesa3d/src/intel/isl/
H A Disl_aux_info.c212 isl_aux_prepare_access(enum isl_aux_state initial_state, in isl_aux_prepare_access()
253 isl_aux_state_transition_aux_op(enum isl_aux_state initial_state, in isl_aux_state_transition_aux_op()
290 isl_aux_state_transition_write(enum isl_aux_state initial_state, in isl_aux_state_transition_write()
/aosp_15_r20/external/perfetto/src/tracing/core/
H A Dshared_memory_arbiter_impl_unittest.cc399 void SetupArbiter(InitialBindingState initial_state) { in SetupArbiter()
411 void EnsureArbiterBoundToEndpoint(InitialBindingState initial_state) { in EnsureArbiterBoundToEndpoint()
418 void TestStartupTracing(InitialBindingState initial_state) { in TestStartupTracing()
552 InitialBindingState initial_state) { in TestAbortStartupTracingForReservation()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A 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
H A 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
H A 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
H A Dconvolutional_recurrent.py295 initial_state=None, argument
903 def call(self, inputs, mask=None, training=None, initial_state=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Drnn.py507 initial_state=None, argument
767 initial_state, argument
1313 initial_state=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dstateless_random_ops.cc87 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniform() local
123 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniformFullInt() local
306 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in Compile() local
H A Dstateful_random_ops.cc68 xla::XlaOp initial_state, in StatefulRngUniform()
94 xla::XlaOp initial_state, in StatefulRngUniformFullInt()
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/spirit/include/boost/spirit/home/lex/lexer/lexertl/
Dstatic_lexer.hpp174 , char_type const* initial_state = 0) const in begin()
219 string_type initial_state() const in initial_state() function in boost::spirit::lex::lexertl::static_lexer
Dlexer.hpp199 , char_type const* initial_state = 0) const in begin()
284 string_type initial_state() const in initial_state() function in boost::spirit::lex::lexertl::lexer
/aosp_15_r20/external/yapf/yapf/yapflib/
H A Dreformatter.py453 def _AnalyzeSolutionSpace(initial_state): argument
541 def _ReconstructPath(initial_state, current): argument
/aosp_15_r20/external/cronet/base/synchronization/
H A Dwaitable_event_posix.cc50 InitialState initial_state) in WaitableEvent()
363 InitialState initial_state) in WaitableEventKernel()
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/spirit/include/boost/spirit/home/lex/
Dtokenize_and_parse.hpp298 , typename Lexer::char_type const* initial_state = 0) in tokenize()
316 , typename Lexer::char_type const* initial_state = 0) in tokenize()
/aosp_15_r20/external/libchrome/base/synchronization/
H A Dwaitable_event_posix.cc46 InitialState initial_state) in WaitableEvent()
373 InitialState initial_state) in WaitableEventKernel()
/aosp_15_r20/external/libchrome/base/threading/
H A Dthread_perftest.cc236 WaitableEvent::InitialState initial_state) in ConditionVariableEvent()
276 WaitableEvent::InitialState initial_state) { in PthreadEvent()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dscan_dataset_op.cc56 std::vector<Tensor> initial_state(initial_state_inputs.begin(), in MakeDataset() local
74 std::vector<Tensor> initial_state, in Dataset()
/aosp_15_r20/external/cronet/base/task/sequence_manager/
H A Dthread_controller.cc160 void ThreadController::RunLevelTracker::OnRunLoopStarted(State initial_state, in OnRunLoopStarted()
284 ThreadController::RunLevelTracker::RunLevel::RunLevel(State initial_state, in RunLevel()
/aosp_15_r20/external/cronet/base/threading/
H A Dthread_perftest.cc261 WaitableEvent::InitialState initial_state) in ConditionVariableEvent()
301 WaitableEvent::InitialState initial_state) { in PthreadEvent()
/aosp_15_r20/external/libopus/dnn/torch/rdovae/rdovae/
H A Drdovae.py436 def forward(self, z, initial_state): argument
681 def decode(self, z, initial_state): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dscan_ops.py22 def scan(initial_state, scan_func): argument
/aosp_15_r20/external/cronet/base/test/
H A Dtest_waitable_event.cc14 InitialState initial_state) in TestWaitableEvent()
/aosp_15_r20/external/pytorch/torch/onnx/_internal/diagnostics/infra/sarif/
H A D_thread_flow.py29 initial_state: Any = dataclasses.field( variable in ThreadFlow
H A D_graph_traversal.py29 initial_state: Any = dataclasses.field( variable in GraphTraversal

1234