Searched defs:compute_dtype (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_micro_gemm.py | 73 compute_dtype, argument
209 compute_dtype=None, argument
253 self, name, input_dtype, input2_dtype, output_dtype, compute_dtype, alpha argument
752 compute_dtype=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| H A D | policy.py | 263 def compute_dtype(self): member in Policy
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | observer.py | 1454 compute_dtype=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | base_layer.py | 2318 def compute_dtype(self): member in Layer
|