Searched defs:weight_tensor (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 163 const Tensor& weight_tensor = in Compute() local
397 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 | 427 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_version.cc | 720 const Tensor* weight_tensor = in GetOpSignature() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model.cc | 142 const TensorT* weight_tensor, TensorT* bias_tensor, in QuantizeBias()
1124 TensorT* weight_tensor = in QuantizeBiases() local
1323 TensorT* weight_tensor = in EnsureBiasScaleCompatibility() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 680 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 | 3806 def _transform_weight_tensor(self, weight_tensor): argument
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3224 const auto weight_tensor = in Map() local
|