Searched defs:weight_tensor (Results 1 – 11 of 11) 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
405 const Tensor& bias_tensor, const Tensor& weight_tensor, in GetBiasHandle()
|
| D | mkl_matmul_op_fused.cc | 64 const Tensor& weight_tensor = ctx->input(this->kInputIndexWeight); in Compute() local
|
| D | mkl_matmul_ops_common.h | 503 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
|
| /external/ComputeLibrary/src/graph/mutators/ |
| D | InPlaceOperationMutator.cpp | 100 auto weight_tensor = weight_edge->tensor(); in try_in_place_depthwiseconv() local
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_signature.cc | 154 const Tensor* weight_tensor = in GetOpSignature() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| 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
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 688 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| D | lstm.cc | 1128 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 3914 def _transform_weight_tensor(self, weight_tensor): argument
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 4053 const auto weight_tensor = in Map() local
|