| /external/grpc-grpc/src/core/lib/backoff/ |
| D | backoff.cc | 33 double generate_uniform_random_number(uint32_t* rng_state) { in generate_uniform_random_number() argument
35 *rng_state = (1103515245 * *rng_state + 12345) % two_raise_31; in generate_uniform_random_number()
36 return *rng_state / static_cast<double>(two_raise_31); in generate_uniform_random_number()
39 double generate_uniform_random_number_between(uint32_t* rng_state, double a, in generate_uniform_random_number_between() argument
44 return a + generate_uniform_random_number(rng_state) * range; in generate_uniform_random_number_between()
|
| /external/rust/crates/grpcio-sys/grpc/src/core/lib/backoff/ |
| D | backoff.cc | 33 double generate_uniform_random_number(uint32_t* rng_state) { in generate_uniform_random_number() argument
35 *rng_state = (1103515245 * *rng_state + 12345) % two_raise_31; in generate_uniform_random_number()
36 return *rng_state / static_cast<double>(two_raise_31); in generate_uniform_random_number()
39 double generate_uniform_random_number_between(uint32_t* rng_state, double a, in generate_uniform_random_number_between() argument
44 return a + generate_uniform_random_number(rng_state) * range; in generate_uniform_random_number_between()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 253 int64* rng_state) { in InitializeTypedBuffer() argument
283 int64& host_index = *rng_state; in InitializeTypedBuffer()
305 int64* rng_state, se::DeviceMemoryBase buffer) { in InitializeBuffer() argument
308 return InitializeTypedBuffer<Eigen::half>(stream, buffer, rng_state); in InitializeBuffer()
311 return InitializeTypedBuffer<float>(stream, buffer, rng_state); in InitializeBuffer()
314 return InitializeTypedBuffer<double>(stream, buffer, rng_state); in InitializeBuffer()
316 return InitializeTypedBuffer<int8>(stream, buffer, rng_state); in InitializeBuffer()
|
| D | gemm_algorithm_picker.cc | 76 int64 rng_state = 0; in DoUncachedGemmAutotune() local
83 InitializeBuffer(stream, op->shape().element_type(), &rng_state, buffer); in DoUncachedGemmAutotune()
124 int64 rng_state = 0; in DoUncachedGemmAutotune() local
125 InitializeBuffer(stream, gemm->shape().element_type(), &rng_state, in DoUncachedGemmAutotune()
|
| D | stream_executor_util.h | 88 int64* rng_state, se::DeviceMemoryBase buffer);
|
| D | gpu_conv_algorithm_picker.cc | 387 int64 rng_state = 0; in PickBestAlgorithmNoCacheCuda() local
394 const auto initialize_buffer = [init_conv_data, &stream, &rng_state]( in PickBestAlgorithmNoCacheCuda()
398 InitializeBuffer(stream, buffer_shape.element_type(), &rng_state, buffer); in PickBestAlgorithmNoCacheCuda()
|
| D | ir_emitter_unnested.h | 199 Status HandleRngGetAndUpdateState(HloInstruction* rng_state) override;
|
| D | ir_emitter_unnested.cc | 2455 HloInstruction* rng_state) { in HandleRngGetAndUpdateState() argument
2456 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(rng_state)); in HandleRngGetAndUpdateState()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | rng_get_and_update_state.hlo | 8 // CHECK: %[[VAL_3:.*]] = load i128, i128* @rng_state, align 16
10 // CHECK: store i128 %[[VAL_4]], i128* @rng_state, align 16
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 172 Status HandleRngGetAndUpdateState(HloInstruction* rng_state) override;
|
| D | ir_emitter.cc | 2758 Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) { in HandleRngGetAndUpdateState() argument
2759 VLOG(2) << "RngGetAndUpdateState: " << rng_state->ToString(); in HandleRngGetAndUpdateState()
2761 Cast<HloRngGetAndUpdateStateInstruction>(rng_state)->delta(), module_, in HandleRngGetAndUpdateState()
2764 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(rng_state)); in HandleRngGetAndUpdateState()
2765 llvm::Value* address = GetEmittedValueFor(rng_state); in HandleRngGetAndUpdateState()
2774 rng_state->shape().element_type()))); in HandleRngGetAndUpdateState()
|