| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cpu_select_algorithm.py | 156 self, batch_size, in_features, out_features, bias, input_3d, dtype argument
191 def test_linear_wgt_multi_users(self, in_features, out_features, bias, dtype): argument
266 self, batch_size, in_features, out_features, bias, epilogue, dtype argument
349 self, batch_size, in_features, out_features, bias, epilogue, dtype argument
397 out_features, argument
483 self, batch_size, in_features, linear_in_features, out_features, bias, dtype argument
566 self, batch_size, in_features, out_features, bias, unary, binary, dtype argument
601 self, batch_size, in_features, out_features, bias, binary, dtype argument
629 def test_linear_amx(self, batch_size, in_features, out_features, bias, dtype): argument
665 out_features, argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/ |
| H A D | linear_relu.py | 35 def __init__(self, in_features, out_features, bias=True, dtype=torch.qint8): argument
76 self, in_features, out_features, negative_slope, bias=True, dtype=torch.qint8 argument
150 def __init__(self, in_features, out_features, bias=True, dtype=torch.qint8): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | linear.cpp | 6 LinearOptions::LinearOptions(int64_t in_features, int64_t out_features) in LinearOptions()
12 int64_t out_features) in BilinearOptions()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | linear.h | 52 LinearImpl(int64_t in_features, int64_t out_features) in LinearImpl()
178 BilinearImpl(int64_t in1_features, int64_t in2_features, int64_t out_features) in BilinearImpl()
|
| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/rl/ |
| H A D | agent.py | 18 def __init__(self, in_features, nlayers, out_features): argument
68 def set_world(self, batch_size, state_size, nlayers, out_features, batch=True): argument
|
| H A D | coordinator.py | 19 def __init__(self, batch_size, batch, state_size, nlayers, out_features): argument
|
| H A D | launcher.py | 59 out_features, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_module.py | 99 batch_size, in_features, out_features, use_bias, argument
1603 … def test_linear_api(self, batch_size, in_features, out_features, use_bias, use_default_observer): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/qat/dynamic/modules/ |
| H A D | linear.py | 24 out_features, argument
|
| /aosp_15_r20/external/executorch/examples/cadence/operators/ |
| H A D | quantized_linear_op.py | 24 out_features = 16 variable
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| H A D | linear_relu.py | 33 def __init__(self, in_features, out_features, bias=True, qconfig=None): argument
|
| H A D | linear_fused.py | 37 out_features, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/ |
| H A D | linear.py | 36 out_features, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/dynamic/modules/ |
| H A D | linear_relu.py | 32 def __init__(self, in_features, out_features, bias=True, dtype=torch.qint8): argument
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| H A D | constraint_generator.py | 1020 def linear_constraints(n: Node, in_features, out_features, symbols, counter): argument
1065 def add_linear_constraints(dims1, dims2, in_features, out_features): argument
|
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A D | test_deprecated_jit_quant.py | 156 def __init__(self, in_features, out_features): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/examples/ |
| H A D | parameter_server_test.py | 24 out_features = 30 variable
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | linear.py | 41 def __init__(self, in_features, out_features, bias_=True, dtype=torch.qint8): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/ |
| H A D | linear.py | 28 out_features, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/consistency_integration_test/ |
| H A D | function_io_tests.py | 188 def __init__(self, in_features, out_features): argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | linear_weight_int4_test.cpp | 31 const int64_t out_features = weights_4x2.size(0); in linear_weight_int4_reference_impl() local
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | linear.py | 151 def __init__(self, in_features, out_features, bias_=True, dtype=torch.qint8): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/ |
| H A D | linear.py | 113 out_features, argument
|
| /aosp_15_r20/external/executorch/examples/models/llama/source_transformation/ |
| H A D | quantize.py | 425 out_features, argument
|
| /aosp_15_r20/external/pytorch/test/higher_order_ops/ |
| H A D | test_with_effects.py | 352 def __init__(self, in_features, out_features): argument
|