| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLFuseBatchNormalizationKernel.cpp | 44 … const ITensorInfo *input_bias, const ITensorInfo *bn_beta, const ITensorInfo *bn_gamma, in validate_arguments()
111 … const ICLTensor *input_bias, const ICLTensor *bn_beta, const ICLTensor *bn_gamma, in configure()
119 … const ICLTensor *input_bias, const ICLTensor *bn_beta, const ICLTensor *bn_gamma, in configure()
184 … const ITensorInfo *input_bias, const ITensorInfo *bn_beta, const ITensorInfo *bn_gamma, in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | quant_lstm_sup_test.cc | 259 std::vector<int32_t> input_bias; in TEST() local
282 std::vector<int32_t> input_bias; in TEST() local
305 std::vector<int32_t> input_bias; in TEST() local
328 std::vector<int32_t> input_bias; in TEST() local
|
| H A D | quant_lstm_sup.cc | 131 std::vector<int32_t>* input_bias, in DecomposeBiasTensor()
|
| H A D | nnapi_delegate.cc | 4127 std::vector<int32_t> input_bias; in Map() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLFuseBatchNormalization.cpp | 46 … const ICLTensor *input_bias, const ICLTensor *bn_beta, const ICLTensor *bn_gamma, in configure()
54 … const ICLTensor *input_bias, const ICLTensor *bn_beta, const ICLTensor *bn_gamma, in configure()
63 … const ITensorInfo *input_bias, const ITensorInfo *bn_beta, const ITensorInfo *bn_gamma, in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NEFuseBatchNormalizationKernel.cpp | 155 … const ITensorInfo *input_bias, const ITensorInfo *bn_beta, const ITensorInfo *bn_gamma, in validate_arguments()
222 … const ITensor *input_bias, const ITensor *bn_beta, const ITensor *bn_gamma, in configure()
273 … const ITensorInfo *input_bias, const ITensorInfo *bn_beta, const ITensorInfo *bn_gamma, in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEFuseBatchNormalization.cpp | 45 … const ITensor *input_bias, const ITensor *bn_beta, const ITensor *bn_gamma, in configure()
57 … const ITensorInfo *input_bias, const ITensorInfo *bn_beta, const ITensorInfo *bn_gamma, in validate()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | RNN.cu | 34 const TensorArg& input_bias, const TensorArg& hidden_bias, in checkSizes()
371 const Tensor& input_bias, const Tensor& hidden_bias, in lstm_forward_impl()
444 const Tensor& input_bias, const Tensor& hidden_bias, in gru_forward_impl()
522 const Tensor& input_bias = *input_bias_maybe_owned; in _thnn_fused_lstm_cell_cuda() local
608 const Tensor& input_bias = *input_bias_maybe_owned; in _thnn_fused_gru_cell_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 1553 const Tensor& input_bias = c10::value_or_else(input_bias_opt, [] {return Tensor();}); in _thnn_differentiable_lstm_cell_backward() local
1605 const Tensor& input_bias = *input_bias_maybe_owned; in _thnn_differentiable_gru_cell_backward() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 5865 input_bias, argument
5906 input_bias=None, argument
|