Home
last modified time | relevance | path

Searched refs:weight_tensor (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py2078 weight_tensor = sparse_tensors.weight_tensor
2079 if weight_tensor is not None:
2080 weight_tensor = sparse_ops.sparse_reshape(
2081 weight_tensor, [array_ops.shape(weight_tensor)[0], -1])
2095 sparse_weights=weight_tensor,
2535 sparse_weights = sparse_tensors.weight_tensor
2657 sparse_weights = sparse_tensors.weight_tensor
3001 weight_tensor = inputs.get(self.weight_feature_key)
3002 if weight_tensor is None:
3004 weight_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
[all …]
Dfeature_column_v2.py2264 weight_tensor = sparse_tensors.weight_tensor
2265 if weight_tensor is not None:
2266 weight_tensor = sparse_ops.sparse_reshape(
2267 weight_tensor, [array_ops.shape(weight_tensor)[0], -1])
2272 sparse_weights=weight_tensor,
2912 sparse_weights = sparse_tensors.weight_tensor
3217 sparse_weights = sparse_tensors.weight_tensor
3806 def _transform_weight_tensor(self, weight_tensor): argument
3807 if weight_tensor is None:
3809 weight_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
[all …]
Dsequence_feature_column_test.py179 self.assertIsNone(id_weight_pair.weight_tensor)
254 self.assertIsNone(id_weight_pair.weight_tensor)
307 self.assertIsNone(id_weight_pair.weight_tensor)
333 self.assertIsNone(id_weight_pair.weight_tensor)
374 self.assertIsNone(id_weight_pair.weight_tensor)
Dfeature_column_v2_test.py616 self.assertIsNone(id_weight_pair.weight_tensor)
634 self.assertIsNone(id_weight_pair.weight_tensor)
946 self.assertIsNone(id_weight_pair.weight_tensor)
957 self.assertIsNone(id_weight_pair.weight_tensor)
1308 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
3213 self.assertIsNone(id_weight_pair.weight_tensor)
3237 self.assertIsNone(id_weight_pair.weight_tensor)
3282 self.assertIsNone(id_weight_pair.weight_tensor)
3308 self.assertIsNone(id_weight_pair.weight_tensor)
3335 self.assertIsNone(id_weight_pair.weight_tensor)
[all …]
Dfeature_column_test.py500 self.assertIsNone(id_weight_pair.weight_tensor)
515 self.assertIsNone(id_weight_pair.weight_tensor)
824 self.assertIsNone(id_weight_pair.weight_tensor)
846 self.assertIsNone(id_weight_pair.weight_tensor)
1162 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
1249 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
3525 self.assertIsNone(id_weight_pair.weight_tensor)
3544 self.assertIsNone(id_weight_pair.weight_tensor)
3601 self.assertIsNone(id_weight_pair.weight_tensor)
3623 self.assertIsNone(id_weight_pair.weight_tensor)
[all …]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_qmatmul_op.cc163 const Tensor& weight_tensor = in Compute() local
186 : weight_tensor.shape(); in Compute()
221 weight.SetUsrMem(weight_md, &weight_tensor); in Compute()
271 weight_tensor, weight, weight_md); in Compute()
279 weight.SetUsrMem(weight_md, &weight_tensor); in Compute()
288 const_cast<Tweight*>(weight_tensor.flat<Tweight>().data())); in Compute()
295 context, matmul_fwd_pd, bias_tensor, weight_tensor, cpu_stream); in Compute()
397 const Tensor& bias_tensor, const Tensor& weight_tensor, in GetBiasHandle() argument
417 int k = weight_tensor.dim_size(0); in GetBiasHandle()
418 int n = weight_tensor.dim_size(1); in GetBiasHandle()
[all …]
Dmkl_matmul_op_fused.cc60 const Tensor& weight_tensor = ctx->input(this->kInputIndexWeight); in Compute() local
73 auto weight_tf_shape = weight_tensor.shape(); in Compute()
204 T* weight_data = const_cast<T*>(weight_tensor.flat<T>().data()); in Compute()
231 this->CacheWeight(ctx, matmul_pd, cached_weight_data, weight_tensor, in Compute()
Dmkl_matmul_ops_common.h427 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight() argument
439 weight.SetUsrMem(weight_md, &weight_tensor); in CacheWeight()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc142 const TensorT* weight_tensor, TensorT* bias_tensor, in QuantizeBias() argument
152 TF_LITE_ENSURE(error_reporter, weight_tensor->quantization); in QuantizeBias()
153 std::vector<float> weight_scales = weight_tensor->quantization->scale; in QuantizeBias()
156 if (bias_tensor->shape[0] != weight_tensor->shape[channel_dim_index]) { in QuantizeBias()
160 bias_tensor->shape[0], weight_tensor->shape[channel_dim_index]); in QuantizeBias()
1124 TensorT* weight_tensor = in QuantizeBiases() local
1129 model, input_tensor, weight_tensor, bias_tensor, in QuantizeBiases()
1323 TensorT* weight_tensor = in EnsureBiasScaleCompatibility() local
1351 if (!utils::HasMinMax(weight_tensor)) { in EnsureBiasScaleCompatibility()
1362 weight_tensor->shape[weight_property.per_axis_index]) { in EnsureBiasScaleCompatibility()
[all …]
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc720 const Tensor* weight_tensor = in GetOpSignature() local
723 (weight_tensor->sparsity() != nullptr); in GetOpSignature()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc3224 const auto weight_tensor = in Map() local
3237 weight_tensor.data.uint8, weight_tensor.dims, &recurrent_to_input, in Map()
3245 weight_tensor.dims, recurrent_weight_dims, input_weight_dims); in Map()
3251 input_weight_dims, input_to_input, weight_tensor.params, in Map()
3256 input_weight_dims, input_to_forget, weight_tensor.params, in Map()
3261 input_weight_dims, input_to_cell, weight_tensor.params, in Map()
3266 input_weight_dims, input_to_output, weight_tensor.params, in Map()
3271 recurrent_weight_dims, recurrent_to_input, weight_tensor.params, in Map()
3276 recurrent_weight_dims, recurrent_to_forget, weight_tensor.params, in Map()
3281 recurrent_weight_dims, recurrent_to_cell, weight_tensor.params, in Map()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc680 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias() argument
682 if (weight_tensor == nullptr) { in PrecomputeZeroPointTimesWeightWithBias()
686 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias()
698 const int8_t* weight = GetTensorData<int8_t>(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias()
Dlstm.cc1127 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias() argument
1129 if (weight_tensor == nullptr) { in PrecomputeZeroPointTimesWeightWithBias()
1133 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias()
1145 const int8_t* weight = GetTensorData<int8_t>(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias()