| /external/pytorch/test/inductor/ |
| 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
395 in_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
663 in_features, argument
[all …]
|
| /external/pytorch/torch/ao/nn/intrinsic/quantized/modules/ |
| 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
|
| /external/pytorch/torch/csrc/api/src/nn/options/ |
| D | adaptive.cpp | 7 int64_t in_features, in AdaptiveLogSoftmaxWithLossOptions()
|
| D | linear.cpp | 6 LinearOptions::LinearOptions(int64_t in_features, int64_t out_features) in LinearOptions()
|
| /external/pytorch/torch/ao/nn/qat/dynamic/modules/ |
| D | linear.py | 23 in_features, argument
|
| /external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| D | adaptive.h | 48 int64_t in_features, in AdaptiveLogSoftmaxWithLossImpl()
|
| D | linear.h | 52 LinearImpl(int64_t in_features, int64_t out_features) in LinearImpl()
|
| /external/executorch/examples/cadence/operators/ |
| D | quantized_linear_op.py | 23 in_features = 32 variable
|
| /external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| D | linear_relu.py | 33 def __init__(self, in_features, out_features, bias=True, qconfig=None): argument
|
| D | linear_fused.py | 36 in_features, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 46 const int64_t in_features = GetTensorDim(tensor_in, data_format, 'C'); in Compute() local
141 const int64_t in_features = GetTensorDim(tensor_in_shape, data_format, 'C'); in Compute() local
|
| /external/pytorch/torch/ao/nn/qat/modules/ |
| D | linear.py | 35 in_features, argument
|
| /external/pytorch/torch/ao/nn/intrinsic/quantized/dynamic/modules/ |
| D | linear_relu.py | 32 def __init__(self, in_features, out_features, bias=True, dtype=torch.qint8): argument
|
| /external/pytorch/test/quantization/jit/ |
| D | test_deprecated_jit_quant.py | 156 def __init__(self, in_features, out_features): argument
|
| /external/pytorch/torch/testing/_internal/distributed/rpc/examples/ |
| D | parameter_server_test.py | 23 in_features = 100 variable
|
| /external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| D | linear.py | 41 def __init__(self, in_features, out_features, bias_=True, dtype=torch.qint8): argument
|
| /external/pytorch/benchmarks/distributed/rpc/rl/ |
| D | agent.py | 18 def __init__(self, in_features, nlayers, out_features): argument
|
| /external/libopus/dnn/ |
| D | lpcnet_demo.c | 174 float in_features[5*NB_TOTAL_FEATURES]; in main() local
|
| /external/pytorch/torch/ao/nn/sparse/quantized/dynamic/ |
| D | linear.py | 27 in_features, argument
|
| /external/tensorflow/tensorflow/tools/consistency_integration_test/ |
| D | function_io_tests.py | 188 def __init__(self, in_features, out_features): argument
|
| /external/pytorch/test/quantization/core/ |
| 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
|
| /external/pytorch/torch/ao/nn/quantized/modules/ |
| D | linear.py | 151 def __init__(self, in_features, out_features, bias_=True, dtype=torch.qint8): argument
|
| /external/pytorch/torch/ao/nn/sparse/quantized/ |
| D | linear.py | 112 in_features, argument
|
| /external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| 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
|
| /external/pytorch/test/higher_order_ops/ |
| D | test_with_effects.py | 352 def __init__(self, in_features, out_features): argument
|