| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | frozen_conv_folding.cpp | 164 bool opDoesNotBroadCastWithConv(Tensor& op_tensor, Tensor& weight_tensor) { in opDoesNotBroadCastWithConv()
189 Tensor weight_tensor = in checkConvAndBroadcastingOpPreConditions() local
256 Tensor weight_tensor = in FoldFrozenConvAddOrSub() local
322 Tensor weight_tensor = in FoldFrozenConvMulOrDiv() local
|
| H A D | frozen_concat_linear.cpp | 69 auto weight_tensor = constant_as<Tensor>(weight).value(); in collectConstantLinearLayers() local
126 Tensor weight_tensor = in mergeLinearLayers() local
|
| H A D | frozen_linear_transpose.cpp | 60 Tensor weight_tensor = constant_as<Tensor>(weight).value(); in replace_linear_with_matmul() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_qmatmul_op.cc | 163 const Tensor& weight_tensor = in Compute() local
405 const Tensor& bias_tensor, const Tensor& weight_tensor, in GetBiasHandle()
|
| H A D | mkl_matmul_op_fused.cc | 64 const Tensor& weight_tensor = ctx->input(this->kInputIndexWeight); in Compute() local
|
| H A D | mkl_matmul_ops_common.h | 503 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | binary_folding.py | 105 def _op_not_broadcasting_with_conv(weight_tensor, other_tensor): argument
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | QLSTMLayerNormalizationFixture.h | 61 void fill(InputType &&input_tensor, InputType &&weight_tensor, BiasType &&bias_tensor) in fill()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | InPlaceOperationMutator.cpp | 100 auto weight_tensor = weight_edge->tensor(); in try_in_place_depthwiseconv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantize_model.cc | 276 const TensorT* weight_tensor, TensorT* bias_tensor, in QuantizeBias()
1586 TensorT* weight_tensor = in QuantizeBiases() local
1797 TensorT* weight_tensor = in EnsureBiasScaleCompatibility() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | op_signature.cc | 154 const Tensor* weight_tensor = in GetOpSignature() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/ |
| H A D | etnaviv_ml.h | 76 struct pipe_resource *weight_tensor; member
|
| /aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/ |
| H A D | test_executor.cpp | 114 auto weight_tensor = subgraph->tensors[weights_index]; in patch_conv2d() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.cpp | 64 bool PrecomputeZeroPointTimesWeightWithBias(int32_t zero_point, const int8_t* weight_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | serializer.py | 1894 def get_optional_bias(self, jit_bias, weight_tensor, transpose=False): argument
2071 weight_tensor, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unidirectional_sequence_lstm.cc | 688 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| H A D | lstm.cc | 1128 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv.cpp | 652 at::Tensor weight_tensor; in apply_impl_xnnp() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/include/pipe/ |
| H A D | p_state.h | 1080 struct pipe_tensor *weight_tensor; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
| H A D | feature_column_v2.py | 3914 def _transform_weight_tensor(self, weight_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate.cc | 4053 const auto weight_tensor = in Map() local
|