Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_qmatmul_op.cc160 const Tensor& weight_tensor = in Compute() local
183 : weight_tensor.shape(); in Compute()
217 weight.SetUsrMem(weight_md, &weight_tensor); in Compute()
265 CacheWeight(context, matmul_fwd_pd, weight_data, weight_tensor, in Compute()
274 weight.SetUsrMem(weight_md, &weight_tensor); in Compute()
283 const_cast<Tweight*>(weight_tensor.flat<Tweight>().data())); in Compute()
288 bias_tensor, weight_tensor); in Compute()
379 const Tensor& bias_tensor, const Tensor& weight_tensor) { in GetBiasHandle() argument
398 int k = weight_tensor.dim_size(0); in GetBiasHandle()
399 int n = weight_tensor.dim_size(1); in GetBiasHandle()
[all …]
Dmkl_matmul_op_fused.cc54 const Tensor& weight_tensor = ctx->input(this->kInputIndexWeight); in Compute() local
67 : weight_tensor.shape(); in Compute()
141 T* weight_data = const_cast<T*>(weight_tensor.flat<T>().data()); in Compute()
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py2053 weight_tensor = sparse_tensors.weight_tensor
2054 if weight_tensor is not None:
2055 weight_tensor = sparse_ops.sparse_reshape(
2056 weight_tensor, [array_ops.shape(weight_tensor)[0], -1])
2070 sparse_weights=weight_tensor,
2510 sparse_weights = sparse_tensors.weight_tensor
2632 sparse_weights = sparse_tensors.weight_tensor
2976 weight_tensor = inputs.get(self.weight_feature_key)
2977 if weight_tensor is None:
2979 weight_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
[all …]
Dfeature_column_v2.py2566 weight_tensor = sparse_tensors.weight_tensor
2567 if weight_tensor is not None:
2568 weight_tensor = sparse_ops.sparse_reshape(
2569 weight_tensor, [array_ops.shape(weight_tensor)[0], -1])
2574 sparse_weights=weight_tensor,
3199 sparse_weights = sparse_tensors.weight_tensor
3500 sparse_weights = sparse_tensors.weight_tensor
4079 def _transform_weight_tensor(self, weight_tensor): argument
4080 if weight_tensor is None:
4082 weight_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
[all …]
Dfeature_column_test.py510 self.assertIsNone(id_weight_pair.weight_tensor)
525 self.assertIsNone(id_weight_pair.weight_tensor)
841 self.assertIsNone(id_weight_pair.weight_tensor)
864 self.assertIsNone(id_weight_pair.weight_tensor)
1189 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
1277 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
3559 self.assertIsNone(id_weight_pair.weight_tensor)
3578 self.assertIsNone(id_weight_pair.weight_tensor)
3635 self.assertIsNone(id_weight_pair.weight_tensor)
3657 self.assertIsNone(id_weight_pair.weight_tensor)
[all …]
Dsequence_feature_column_test.py760 self.assertIsNone(id_weight_pair.weight_tensor)
835 self.assertIsNone(id_weight_pair.weight_tensor)
888 self.assertIsNone(id_weight_pair.weight_tensor)
914 self.assertIsNone(id_weight_pair.weight_tensor)
955 self.assertIsNone(id_weight_pair.weight_tensor)
Dfeature_column_v2_test.py662 self.assertIsNone(id_weight_pair.weight_tensor)
680 self.assertIsNone(id_weight_pair.weight_tensor)
1058 self.assertIsNone(id_weight_pair.weight_tensor)
1070 self.assertIsNone(id_weight_pair.weight_tensor)
1436 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
1547 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
4113 self.assertIsNone(id_weight_pair.weight_tensor)
4138 self.assertIsNone(id_weight_pair.weight_tensor)
4185 self.assertIsNone(id_weight_pair.weight_tensor)
4212 self.assertIsNone(id_weight_pair.weight_tensor)
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc65 const TensorT* weight_tensor, TensorT* bias_tensor, in QuantizeBias() argument
74 TF_LITE_ENSURE(error_reporter, weight_tensor->quantization); in QuantizeBias()
75 std::vector<float> weight_scales = weight_tensor->quantization->scale; in QuantizeBias()
78 if (bias_tensor->shape[0] != weight_tensor->shape[channel_dim_index]) { in QuantizeBias()
81 bias_tensor->shape[0], weight_tensor->shape[channel_dim_index]); in QuantizeBias()
876 TensorT* weight_tensor = in QuantizeBiases() local
881 QuantizeBias(model, input_tensor, weight_tensor, bias_tensor, in QuantizeBiases()
1036 TensorT* weight_tensor = in EnsureBiasScaleCompatibility() local
1064 if (!utils::HasMinMax(weight_tensor)) { in EnsureBiasScaleCompatibility()
1074 weight_tensor->shape[weight_property.per_axis_index]) { in EnsureBiasScaleCompatibility()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc2703 const auto weight_tensor = in Map() local
2716 weight_tensor.data.uint8, weight_tensor.dims, &recurrent_to_input, in Map()
2724 weight_tensor.dims, recurrent_weight_dims, input_weight_dims); in Map()
2730 input_weight_dims, input_to_input, weight_tensor.params, in Map()
2735 input_weight_dims, input_to_forget, weight_tensor.params, in Map()
2740 input_weight_dims, input_to_cell, weight_tensor.params, in Map()
2745 input_weight_dims, input_to_output, weight_tensor.params, in Map()
2750 recurrent_weight_dims, recurrent_to_input, weight_tensor.params, in Map()
2755 recurrent_weight_dims, recurrent_to_forget, weight_tensor.params, in Map()
2760 recurrent_weight_dims, recurrent_to_cell, weight_tensor.params, in Map()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dlstm.cc643 const TfLiteTensor* weight_tensor, const TfLiteTensor* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias() argument
645 if (weight_tensor == nullptr) { in PrecomputeZeroPointTimesWeightWithBias()
649 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias()
661 const int8_t* weight = GetTensorData<int8_t>(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias()