Home
last modified time | relevance | path

Searched defs:rng_state (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_prims/
H A Drng_prims.py223 def __call__(self, rng_state, op, *args, **kwargs): argument
233 def impl_cuda(rng_state, op, *args, **kwargs): argument
241 def impl_cpu(rng_state, op, *args, **kwargs): argument
249 def impl_hpu(rng_state, op, *args, **kwargs): argument
259 def impl_xpu(rng_state, op, *args, **kwargs): argument
267 def impl_proxy_dispatch_mode(mode, rng_state, op, *args, **kwargs): argument
280 def impl_backend_select(rng_state, op, *args, **kwargs): argument
293 def impl_fake_tensor_mode(mode, rng_state, op, *args, **kwargs): argument
300 def impl_functional(ctx, rng_state, op, *args, **kwargs): argument
/aosp_15_r20/external/openthread/third_party/mbedtls/repo/tests/src/
H A Drandom.c29 int mbedtls_test_rnd_std_rand(void *rng_state, in mbedtls_test_rnd_std_rand()
54 int mbedtls_test_rnd_zero_rand(void *rng_state, in mbedtls_test_rnd_zero_rand()
67 int mbedtls_test_rnd_buffer_rand(void *rng_state, in mbedtls_test_rnd_buffer_rand()
102 int mbedtls_test_rnd_pseudo_rand(void *rng_state, in mbedtls_test_rnd_pseudo_rand()
/aosp_15_r20/external/mbedtls/tests/src/
H A Drandom.c29 int mbedtls_test_rnd_std_rand(void *rng_state, in mbedtls_test_rnd_std_rand()
54 int mbedtls_test_rnd_zero_rand(void *rng_state, in mbedtls_test_rnd_zero_rand()
67 int mbedtls_test_rnd_buffer_rand(void *rng_state, in mbedtls_test_rnd_buffer_rand()
102 int mbedtls_test_rnd_pseudo_rand(void *rng_state, in mbedtls_test_rnd_pseudo_rand()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_algorithm_picker.cc80 int64_t& rng_state) { in CreateBuffer()
124 int64_t rng_state = 0; in GetBestAlgorithm() local
256 int64_t rng_state = 0; in DoGemmAutotune() local
H A Dstream_executor_util.cc399 int64_t* rng_state) { in InitializeTypedBuffer()
451 int64_t* rng_state, se::DeviceMemoryBase buffer) { in InitializeBuffer()
H A Dgpu_conv_algorithm_picker.cc650 int64_t rng_state = 0; in PickBestAlgorithmNoCacheCuda() local
/aosp_15_r20/hardware/interfaces/rebootescrow/aidl/default/
H A DHadamardUtils.cpp81 uint64_t rng_state = RNG_INV_SEED; in EncodeKey() local
160 uint64_t rng_state = RNG_SEED; in DecodeKey() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DCPUGeneratorImpl.cpp177 auto rng_state = (CPUGeneratorImplState*)new_state.data(); in set_state() local
226 auto rng_state = state_tensor.data_ptr(); in get_state() local
/aosp_15_r20/external/cronet/net/third_party/quiche/src/quiche/common/
H A Dquiche_random.cc27 static thread_local uint64_t rng_state[4] = { in Xoshiro256PlusPlus() local
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/
H A DXPUGeneratorImpl.cpp111 auto rng_state = state_tensor.data_ptr<uint8_t>(); in get_state() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAGeneratorImpl.cpp330 auto rng_state = state_tensor.data_ptr<uint8_t>(); in get_state() local
/aosp_15_r20/external/libxml2/
H A Dtestdict.c455 rng_state[2] = { 123, 456 }; variable
/aosp_15_r20/external/cronet/third_party/libxml/src/
H A Dtestdict.c455 rng_state[2] = { 123, 456 }; variable
/aosp_15_r20/external/mbedtls/programs/test/
H A Dbenchmark.c445 static int myrand(void *rng_state, unsigned char *output, size_t len) in myrand()
/aosp_15_r20/external/openthread/third_party/mbedtls/repo/programs/test/
H A Dbenchmark.c452 static int myrand(void *rng_state, unsigned char *output, size_t len) in myrand()
/aosp_15_r20/hardware/google/gfxstream/third-party/astc-encoder/Source/
Dastcenc_block_sizes.cpp729 uint64_t rng_state[2]; in assign_kmeans_texels() local
/aosp_15_r20/external/mbedtls/library/
H A Drsa.c2488 static int myrand(void *rng_state, unsigned char *output, size_t len) in myrand()
H A Dpsa_crypto.c98 uint8_t rng_state; member
/aosp_15_r20/external/openthread/third_party/mbedtls/repo/library/
H A Drsa.c2889 static int myrand(void *rng_state, unsigned char *output, size_t len) in myrand()
H A Dpsa_crypto.c117 uint8_t rng_state; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc2939 Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) { in HandleRngGetAndUpdateState()