Home
last modified time | relevance | path

Searched defs:float (Results 1 – 25 of 1470) sorted by relevance

12345678910>>...59

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding.py371 learning_rate: float,
373 clip_weight_min: Optional[float],
374 clip_weight_max: Optional[float],
375 weight_decay_factor: Optional[float],
377 clip_gradient_min: Optional[float] = None,
378 clip_gradient_max: Optional[float] = None,
418 learning_rate: float,
419 initial_accumulator: float = 0.1,
421 clip_weight_min: Optional[float] = None,
422 clip_weight_max: Optional[float] = None,
[all …]
H A Dtpu_embedding_v2_utils.py49 learning_rate: Union[float, Callable[[], float]], argument
51 clip_weight_min: Optional[float],
52 clip_weight_max: Optional[float],
53 weight_decay_factor: Optional[float],
224 learning_rate: Union[float, Callable[[], float]] = 0.01, argument
226 clip_weight_min: Optional[float] = None,
227 clip_weight_max: Optional[float] = None,
228 weight_decay_factor: Optional[float] = None,
326 learning_rate: Union[float, Callable[[], float]] = 0.001, argument
327 initial_accumulator_value: float = 0.1,
[all …]
/aosp_15_r20/external/pytorch/torch/optim/
H A D_adafactor.py25 lr: Union[float, Tensor] = 1e-2,
26 beta2_decay: float = -0.8,
27 eps: Tuple[Optional[float], float] = (None, 1e-3),
28 d: float = 1.0,
29 weight_decay: float = 0.0,
340 d: float,
341 lr: Union[Tensor, float], argument
342 beta2_decay: float,
343 weight_decay: float,
344 eps1: Optional[float],
[all …]
H A Dnadam.py34 lr: Union[float, Tensor] = 2e-3,
35 betas: Tuple[float, float] = (0.9, 0.999), argument
36 eps: float = 1e-8,
37 weight_decay: float = 0,
38 momentum_decay: float = 4e-3,
287 beta1: float,
288 beta2: float,
289 lr: float,
290 weight_decay: float,
291 momentum_decay: float,
[all …]
H A Dadamw.py36 lr: Union[float, Tensor] = 1e-3,
37 betas: Tuple[float, float] = (0.9, 0.999), argument
38 eps: float = 1e-8,
39 weight_decay: float = 1e-2,
328 beta1: float,
329 beta2: float,
330 lr: Union[Tensor, float], argument
331 weight_decay: float,
332 eps: float,
447 beta1: float,
[all …]
H A Dadam.py36 lr: Union[float, Tensor] = 1e-3,
37 betas: Tuple[float, float] = (0.9, 0.999), argument
38 eps: float = 1e-8,
39 weight_decay: float = 0,
332 beta1: float,
333 beta2: float,
334 lr: Union[float, Tensor],
335 weight_decay: float,
336 eps: float,
451 beta1: float,
[all …]
H A Dadamax.py32 lr: Union[float, Tensor] = 2e-3,
33 betas: Tuple[float, float] = (0.9, 0.999), argument
34 eps: float = 1e-8,
35 weight_decay: float = 0,
233 eps: float,
234 beta1: float,
235 beta2: float,
236 lr: float,
237 weight_decay: float,
307 eps: float,
[all …]
H A Dasgd.py32 lr: Union[float, Tensor] = 1e-2,
33 lambd: float = 1e-4,
34 alpha: float = 0.75,
35 t0: float = 1e6,
36 weight_decay: float = 0,
203 lambd: float,
204 lr: float,
205 t0: float,
206 alpha: float,
207 weight_decay: float,
[all …]
H A Dradam.py33 lr: Union[float, Tensor] = 1e-3,
34 betas: Tuple[float, float] = (0.9, 0.999), argument
35 eps: float = 1e-8,
36 weight_decay: float = 0,
261 beta1: float,
262 beta2: float,
263 lr: float,
264 weight_decay: float,
265 eps: float,
360 beta1: float,
[all …]
H A Dsgd.py28 lr: Union[float, Tensor] = 1e-3,
29 momentum: float = 0,
30 dampening: float = 0,
31 weight_decay: float = 0,
256 weight_decay: float,
257 momentum: float,
258 lr: float,
259 dampening: float,
321 weight_decay: float,
322 momentum: float,
[all …]
H A Drmsprop.py32 lr: Union[float, Tensor] = 1e-2,
33 alpha: float = 0.99,
34 eps: float = 1e-8,
35 weight_decay: float = 0,
36 momentum: float = 0,
271 lr: float,
272 alpha: float,
273 eps: float,
274 weight_decay: float,
275 momentum: float,
[all …]
H A Dadagrad.py29 lr: Union[float, Tensor] = 1e-2,
30 lr_decay: float = 0,
31 weight_decay: float = 0,
32 initial_accumulator_value: float = 0,
33 eps: float = 1e-10,
257 lr: float,
258 weight_decay: float,
259 lr_decay: float,
260 eps: float,
329 lr: float,
[all …]
H A Drprop.py32 lr: Union[float, Tensor] = 1e-2,
33 etas: Tuple[float, float] = (0.5, 1.2), argument
34 step_sizes: Tuple[float, float] = (1e-6, 50),
229 step_size_min: float,
230 step_size_max: float,
231 etaminus: float,
232 etaplus: float,
297 step_size_min: float,
298 step_size_max: float,
299 etaminus: float,
[all …]
H A Dadadelta.py31 lr: Union[float, Tensor] = 1.0,
32 rho: float = 0.9,
33 eps: float = 1e-6,
34 weight_decay: float = 0,
252 lr: float,
253 rho: float,
254 eps: float,
255 weight_decay: float,
306 lr: float,
307 rho: float,
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDASparseBlas.cpp11 CUSPARSE_CSRGEAM2_BUFFERSIZE_ARGTYPES(float)) { in csrgeam2_bufferSizeExt()
63 CUSPARSE_CSRGEAM2_BUFFERSIZE_ARGTYPES(c10::complex<float>)) { in csrgeam2_bufferSizeExt()
114 void csrgeam2<float>(CUSPARSE_CSRGEAM2_ARGTYPES(float)) { in csrgeam2()
165 CUSPARSE_CSRGEAM2_ARGTYPES(c10::complex<float>)) { in csrgeam2()
216 void bsrmm<float>(CUSPARSE_BSRMM_ARGTYPES(float)) { in bsrmm()
264 void bsrmm<c10::complex<float>>(CUSPARSE_BSRMM_ARGTYPES(c10::complex<float>)) { in bsrmm()
313 void bsrmv<float>(CUSPARSE_BSRMV_ARGTYPES(float)) { in bsrmv()
353 void bsrmv<c10::complex<float>>(CUSPARSE_BSRMV_ARGTYPES(c10::complex<float>)) { in bsrmv()
396 void bsrsv2_bufferSize<float>(CUSPARSE_BSRSV2_BUFFER_ARGTYPES(float)) { in bsrsv2_bufferSize()
431 CUSPARSE_BSRSV2_BUFFER_ARGTYPES(c10::complex<float>)) { in bsrsv2_bufferSize()
[all …]
/aosp_15_r20/external/fonttools/Lib/fontTools/pens/
H A DttGlyphPen.py60 transformation: Tuple[float, float, float, float, float, float], argument
80 transformation: Tuple[float, float, float, float, float, float], argument
135 round: Callable[[float], int] = otRound,
192 def _addPoint(self, pt: Tuple[float, float], tp: int) -> None: argument
205 def lineTo(self, pt: Tuple[float, float]) -> None: argument
208 def moveTo(self, pt: Tuple[float, float]) -> None: argument
312 pt: Tuple[float, float], argument
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py440 output_ratio: Optional[BroadcastingList2[float]] = None,
517 output_ratio: Optional[BroadcastingList2[float]] = None,
552 output_ratio: Optional[BroadcastingList3[float]] = None,
633 output_ratio: Optional[BroadcastingList3[float]] = None,
1078 norm_type: Union[int, float], argument
1116 norm_type: Union[int, float], argument
1152 norm_type: Union[int, float], argument
1403 p: float = 0.5,
1431 p: float = 0.5,
1454 p: float = 0.5,
[all …]
/aosp_15_r20/external/pytorch/torch/testing/
H A D_comparison.py86 dtype_precisions: Optional[Dict[torch.dtype, Tuple[float, float]]] = None, argument
112 rtol: Optional[float],
113 atol: Optional[float],
147 abs_diff: float,
149 atol: float,
150 rel_diff: float,
152 rtol: float,
176 diff: float,
178 tol: float,
205 actual: Union[bool, int, float, complex], argument
[all …]
H A D_creation.py35 def _uniform_random_(t: torch.Tensor, low: float, high: float) -> torch.Tensor:
48 low: Optional[float] = None,
49 high: Optional[float] = None,
127 low: Optional[float],
128 high: Optional[float],
130 lowest_inclusive: float,
131 highest_exclusive: float,
132 default_low: float,
133 default_high: float,
140 def clamp(a: float, l: float, h: float) -> float:
/aosp_15_r20/external/pytorch/aten/src/ATen/mkl/
H A DSparseBlas.cpp24 void create_csr<float>(MKL_SPARSE_CREATE_CSR_ARGTYPES(float)) { in create_csr()
35 MKL_SPARSE_CREATE_CSR_ARGTYPES(c10::complex<float>)) { in create_csr()
61 void create_bsr<float>(MKL_SPARSE_CREATE_BSR_ARGTYPES(float)) { in create_bsr()
90 MKL_SPARSE_CREATE_BSR_ARGTYPES(c10::complex<float>)) { in create_bsr()
120 void mv<float>(MKL_SPARSE_MV_ARGTYPES(float)) { in mv()
130 void mv<c10::complex<float>>(MKL_SPARSE_MV_ARGTYPES(c10::complex<float>)) { in mv()
153 void add<float>(MKL_SPARSE_ADD_ARGTYPES(float)) { in add()
161 void add<c10::complex<float>>(MKL_SPARSE_ADD_ARGTYPES(c10::complex<float>)) { in add()
172 void export_csr<float>(MKL_SPARSE_EXPORT_CSR_ARGTYPES(float)) { in export_csr()
183 MKL_SPARSE_EXPORT_CSR_ARGTYPES(c10::complex<float>)) { in export_csr()
[all …]
/aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1beta1/
H A Devaluation_service.proto261 optional float score = 1 [(google.api.field_behavior) = OUTPUT_ONLY]; field
296 optional float score = 1 [(google.api.field_behavior) = OUTPUT_ONLY]; field
340 optional float score = 1 [(google.api.field_behavior) = OUTPUT_ONLY]; field
367 optional float score = 1 [(google.api.field_behavior) = OUTPUT_ONLY]; field
373 optional float confidence = 3 [(google.api.field_behavior) = OUTPUT_ONLY]; field
400 optional float score = 1 [(google.api.field_behavior) = OUTPUT_ONLY]; field
406 optional float confidence = 3 [(google.api.field_behavior) = OUTPUT_ONLY]; field
433 optional float score = 1 [(google.api.field_behavior) = OUTPUT_ONLY]; field
439 optional float confidence = 3 [(google.api.field_behavior) = OUTPUT_ONLY]; field
470 optional float score = 1 [(google.api.field_behavior) = OUTPUT_ONLY]; field
[all …]
/aosp_15_r20/external/webrtc/modules/audio_processing/ns/
H A Dnoise_suppressor.cc72 void ApplyFilterBankWindow(rtc::ArrayView<float, kFftSize> x) { in ApplyFilterBankWindow()
84 void FormExtendedFrame(rtc::ArrayView<const float, kNsFrameSize> frame, in FormExtendedFrame()
85 rtc::ArrayView<float, kFftSize - kNsFrameSize> old_data, in FormExtendedFrame()
95 void OverlapAndAdd(rtc::ArrayView<const float, kFftSize> extended_frame, in OverlapAndAdd()
96 rtc::ArrayView<float, kOverlapSize> overlap_memory, in OverlapAndAdd()
109 void DelaySignal(rtc::ArrayView<const float, kNsFrameSize> frame, in DelaySignal()
110 rtc::ArrayView<float, kFftSize - kNsFrameSize> delay_buffer, in DelaySignal()
122 float ComputeEnergyOfExtendedFrame(rtc::ArrayView<const float, kFftSize> x) { in ComputeEnergyOfExtendedFrame()
133 rtc::ArrayView<const float, kNsFrameSize> frame, in ComputeEnergyOfExtendedFrame()
134 rtc::ArrayView<float, kFftSize - kNsFrameSize> old_data) { in ComputeEnergyOfExtendedFrame()
[all …]
/aosp_15_r20/external/webrtc/modules/audio_processing/agc2/rnn_vad/
H A Dspectral_features.cc30 rtc::ArrayView<const float, kNumBands> new_cepstral_coeffs, in UpdateCepstralDifferenceStats()
31 const RingBuffer<float, kNumBands, kCepstralCoeffsHistorySize>& ring_buf, in UpdateCepstralDifferenceStats()
32 SymmetricMatrixBuffer<float, kCepstralCoeffsHistorySize>* sym_matrix_buf) { in UpdateCepstralDifferenceStats()
67 rtc::ArrayView<const float, kFrameSize20ms24kHz> frame, in ComputeWindowedForwardFft()
68 const std::array<float, kFrameSize20ms24kHz / 2>& half_window, in ComputeWindowedForwardFft()
105 rtc::ArrayView<const float, kFrameSize20ms24kHz> reference_frame, in CheckSilenceComputeFeatures()
107 rtc::ArrayView<float, kNumBands - kNumLowerBands> higher_bands_cepstrum, in CheckSilenceComputeFeatures()
157 rtc::ArrayView<float, kNumLowerBands> average, in ComputeAvgAndDerivatives()
177 rtc::ArrayView<float, kNumLowerBands> bands_cross_corr) { in ComputeNormalizedCepstralCorrelation()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/
H A DcutlassF_f32_aligned.cu12 AttentionKernel<float, cutlass::arch::Sm50, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
31 AttentionKernel<float, cutlass::arch::Sm70, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
50 AttentionKernel<float, cutlass::arch::Sm75, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
69 AttentionKernel<float, cutlass::arch::Sm80, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
88 AttentionKernel<float, cutlass::arch::Sm50, true, 32, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
107 AttentionKernel<float, cutlass::arch::Sm70, true, 32, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
126 AttentionKernel<float, cutlass::arch::Sm75, true, 32, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
145 AttentionKernel<float, cutlass::arch::Sm80, true, 64, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
164 AttentionKernel<float, cutlass::arch::Sm50, true, 32, 128, 65536, true, true>::kNumThreads, in __launch_bounds__()
183 AttentionKernel<float, cutlass::arch::Sm70, true, 32, 128, 65536, true, true>::kNumThreads, in __launch_bounds__()
[all …]
/aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/
H A Dtimer.py43 timer: Callable[[], float], argument
190 timer: Callable[[], float] = timer, argument
285 def autorange(self, callback: Optional[Callable[[int, float], NoReturn]] = None) -> None: argument
291 time_hook: Callable[[], float], argument
292 stop_hook: Callable[[List[float]], bool],
293 min_run_time: float,
294 max_run_time: Optional[float] = None,
312 def _estimate_block_size(self, min_run_time: float) -> int:
333 callback: Optional[Callable[[int, float], NoReturn]] = None, argument
334 min_run_time: float = 0.2,
[all …]

12345678910>>...59