Home
last modified time | relevance | path

Searched defs:in_features (Results 1 – 25 of 35) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cpu_select_algorithm.py156 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 …]
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/
H A Dlinear_relu.py35 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/out/soong/.intermediates/hardware/interfaces/biometrics/face/aidl/android.hardware.biometrics.face-V4-ndk-source/gen/android/hardware/biometrics/face/
DISession.cpp76 std::vector<::aidl::android::hardware::biometrics::face::Feature> in_features; in _aidl_android_hardware_biometrics_face_ISession_onTransact() local
260 std::vector<::aidl::android::hardware::biometrics::face::Feature> in_features; in _aidl_android_hardware_biometrics_face_ISession_onTransact() local
491 …cs::face::EnrollmentType in_type, const std::vector<::aidl::android::hardware::biometrics::face::F… in enroll()
947 …cs::face::EnrollmentType in_type, const std::vector<::aidl::android::hardware::biometrics::face::F… in enrollWithContext()
DISessionCallback.cpp208 std::vector<::aidl::android::hardware::biometrics::face::Feature> in_features; in _aidl_android_hardware_biometrics_face_ISessionCallback_onTransact() local
803 …sionCallback::onFeaturesRetrieved(const std::vector<::aidl::android::hardware::biometrics::face::F… in onFeaturesRetrieved()
/aosp_15_r20/out/soong/.intermediates/hardware/interfaces/biometrics/face/aidl/android.hardware.biometrics.face-V4-ndk-source/gen/include/aidl/android/hardware/biometrics/face/
DBnSession.h58 …cs::face::EnrollmentType in_type, const std::vector<::aidl::android::hardware::biometrics::face::F… in enroll()
94 …cs::face::EnrollmentType in_type, const std::vector<::aidl::android::hardware::biometrics::face::F… in enrollWithContext()
DBnSessionCallback.h88 …ScopedAStatus onFeaturesRetrieved(const std::vector<::aidl::android::hardware::biometrics::face::F… in onFeaturesRetrieved()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_pooling_gpu.cc46 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
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dadaptive.cpp7 int64_t in_features, in AdaptiveLogSoftmaxWithLossOptions()
H A Dlinear.cpp6 LinearOptions::LinearOptions(int64_t in_features, int64_t out_features) in LinearOptions()
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_module.py99 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 Dlinear.py23 in_features, argument
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dadaptive.h48 int64_t in_features, in AdaptiveLogSoftmaxWithLossImpl()
H A Dlinear.h52 LinearImpl(int64_t in_features, int64_t out_features) in LinearImpl()
/aosp_15_r20/external/executorch/examples/cadence/operators/
H A Dquantized_linear_op.py23 in_features = 32 variable
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/
H A Dlinear_relu.py33 def __init__(self, in_features, out_features, bias=True, qconfig=None): argument
H A Dlinear_fused.py36 in_features, argument
/aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/
H A Dlinear.py35 in_features, argument
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/dynamic/modules/
H A Dlinear_relu.py32 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 Dconstraint_generator.py1020 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 Dtest_deprecated_jit_quant.py156 def __init__(self, in_features, out_features): argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/examples/
H A Dparameter_server_test.py23 in_features = 100 variable
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Dlinear.py41 def __init__(self, in_features, out_features, bias_=True, dtype=torch.qint8): argument
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/rl/
H A Dagent.py18 def __init__(self, in_features, nlayers, out_features): argument
/aosp_15_r20/external/libopus/dnn/
H A Dlpcnet_demo.c174 float in_features[5*NB_TOTAL_FEATURES]; in main() local
/aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/
H A Dlinear.py27 in_features, argument

12