Searched defs:target_dtype (Results 1 – 10 of 10) sorted by relevance
| /external/pytorch/benchmarks/gpt_fast/ |
| D | mixtral_moe_quantize.py | 12 def dynamically_quantize_per_channel(x, quant_min, quant_max, target_dtype): argument
132 target_dtype=None, argument
149 def __init__(self, num_experts, intermediate_size, dim, target_dtype): argument
|
| D | quantize.py | 10 def dynamically_quantize_per_channel(x, quant_min, quant_max, target_dtype): argument
|
| /external/pytorch/torch/_numpy/ |
| D | _util.py | 140 def typecast_tensor(t, target_dtype, casting): argument
172 def typecast_tensors(tensors, target_dtype, casting): argument
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | rejection_resample_test.py | 154 def testOtherDtypes(self, target_dtype, init_dtype): argument
|
| /external/executorch/examples/models/llama/source_transformation/ |
| D | quantize.py | 180 target_dtype, argument
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 394 target_dtype, argument
|
| /external/pytorch/aten/src/ATen/ |
| D | TensorIterator.h | 151 ScalarType target_dtype = ScalarType::Undefined; member
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_quantization.py | 498 def _dynamically_quantize_per_channel(x, quant_min, quant_max, target_dtype): argument
|
| /external/pytorch/test/ |
| D | test_nn.py | 11606 def compute_result_and_gradient(reduction, target_dtype): argument
|
| D | test_mps.py | 8681 def compute_result_and_gradient(reduction, target_dtype): argument
|