Searched defs:weight_tensor (Results 1 – 10 of 10) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | QLSTMLayerNormalizationFixture.h | 61 void fill(InputType &&input_tensor, InputType &&weight_tensor, BiasType &&bias_tensor) in fill()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 163 const Tensor& weight_tensor = in Compute() local
400 const Tensor& bias_tensor, const Tensor& weight_tensor, in GetBiasHandle()
|
| D | mkl_matmul_op_fused.cc | 60 const Tensor& weight_tensor = ctx->input(this->kInputIndexWeight); in Compute() local
|
| D | mkl_matmul_ops_common.h | 445 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_signature.cc | 142 const Tensor* weight_tensor = in GetOpSignature() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model.cc | 152 const TensorT* weight_tensor, TensorT* bias_tensor, in QuantizeBias()
1143 TensorT* weight_tensor = in QuantizeBiases() local
1350 TensorT* weight_tensor = in EnsureBiasScaleCompatibility() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 687 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| D | lstm.cc | 1127 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 3917 def _transform_weight_tensor(self, weight_tensor): argument
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3474 const auto weight_tensor = in Map() local
|