Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize.py94 if layer_match.weight_tensor is not None:
98 layer_match.weight_tensor.op,
99 input_to_ops_map.ConsumerOperations(layer_match.weight_tensor.op),
486 weight_tensor = match_result.get_tensor(weight_identity_pattern)
487 if weight_tensor is None:
488 weight_tensor = match_result.get_tensor(weight_resource_var_pattern)
489 if weight_tensor is None:
490 weight_tensor = match_result.get_tensor(folded_weight_pattern)
501 _LayerMatch(layer_op, weight_tensor, activation_op, bypass_op,
509 weight_tensor = match_result.get_tensor(weight_identity_pattern)
[all …]
Dfold_batch_norms.py102 weights = match.weight_tensor
105 match.weight_tensor.get_shape().as_list()[2],
106 match.weight_tensor.get_shape().as_list()[3]
258 weight_tensor = match_result.get_tensor(weight_pattern)
311 weight_tensor=weight_tensor,
446 def _CloneWithNewOperands(layer_op, input_tensor, weight_tensor, argument
453 weight_tensor,
462 weight_tensor,
471 weight_tensor,
703 weight_tensor=None,
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dcustom_loss_head.py53 def loss_wrapper(labels, logits, weight_tensor): argument
54 if weight_tensor is None:
55 weight_tensor = array_ops.ones(
57 weighted_loss, _ = loss_fn(labels, weight_tensor, logits)
59 return average_loss, average_loss / math_ops.reduce_mean(weight_tensor)
Ddnn_tree_combined_estimator.py278 weight_tensor = head_lib._weight_tensor(features, head._weight_column)
280 weight_tensor = head_lib._weight_tensor(
293 dnn_logits_fixed, tree_logits, weight_tensor)
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column.py419 def weight_tensor(self, input_tensor): member in _SparseColumn
437 weight_tensor=self.weight_tensor(input_tensor),
497 self.id_tensor(input_tensor), self.weight_tensor(input_tensor))
815 def weight_tensor(self, input_tensor): member in _WeightedSparseColumn
833 weight_tensor=self.weight_tensor(input_tensor),
838 def _do_transform(self, id_tensor, weight_tensor): argument
839 if not isinstance(weight_tensor, sparse_tensor_py.SparseTensor):
841 weight_tensor = contrib_sparse_ops.dense_to_sparse_tensor(weight_tensor)
843 weight_tensor = math_ops.cast(weight_tensor, dtypes.float32)
844 return tuple([id_tensor, weight_tensor])
[all …]
Dtarget_column.py196 def _weighted_loss(self, loss, weight_tensor): argument
201 weight_tensor, shape=(-1,)))
228 weight_tensor = self.get_weight_tensor(features)
229 if weight_tensor is None:
231 loss_weighted = self._weighted_loss(loss_unweighted, weight_tensor)
253 weight_tensor = self.get_weight_tensor(features)
254 if weight_tensor is None:
256 loss_weighted = self._weighted_loss(loss_unweighted, weight_tensor)
259 math_ops.cast(math_ops.reduce_sum(weight_tensor), dtypes.float32),
Dfeature_column_ops.py306 sparse_weights=embedding_lookup_arguments.weight_tensor,
319 assert arg.weight_tensor is None, (
Dfeature_column_test.py370 id_tensor, weight_tensor = _sparse_id_tensor_with_weights(
374 (id_tensor, weight_tensor),
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py1995 weight_tensor = sparse_tensors.weight_tensor
1996 if weight_tensor is not None:
1997 weight_tensor = sparse_ops.sparse_reshape(
1998 weight_tensor, [array_ops.shape(weight_tensor)[0], -1])
2012 sparse_weights=weight_tensor,
2429 sparse_weights = sparse_tensors.weight_tensor
2544 sparse_weights = sparse_tensors.weight_tensor
2892 weight_tensor = inputs.get(self.weight_feature_key)
2893 if weight_tensor is None:
2895 weight_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
[all …]
Dfeature_column_v2.py2450 weight_tensor = sparse_tensors.weight_tensor
2451 if weight_tensor is not None:
2452 weight_tensor = sparse_ops.sparse_reshape(
2453 weight_tensor, [array_ops.shape(weight_tensor)[0], -1])
2458 sparse_weights=weight_tensor,
3052 sparse_weights = sparse_tensors.weight_tensor
3325 sparse_weights = sparse_tensors.weight_tensor
3908 def _transform_weight_tensor(self, weight_tensor): argument
3909 if weight_tensor is None:
3911 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.py748 self.assertIsNone(id_weight_pair.weight_tensor)
787 self.assertIsNone(id_weight_pair.weight_tensor)
840 self.assertIsNone(id_weight_pair.weight_tensor)
866 self.assertIsNone(id_weight_pair.weight_tensor)
907 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)
1006 self.assertIsNone(id_weight_pair.weight_tensor)
1018 self.assertIsNone(id_weight_pair.weight_tensor)
1384 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
1495 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
4709 self.assertIsNone(id_weight_pair.weight_tensor)
4734 self.assertIsNone(id_weight_pair.weight_tensor)
4781 self.assertIsNone(id_weight_pair.weight_tensor)
4808 self.assertIsNone(id_weight_pair.weight_tensor)
[all …]
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_optimizer.py200 weight_tensor = array_ops.reshape(
201 column.weight_tensor(transformed_tensor).values, [-1])
204 weight_tensor = array_ops.ones(
216 weight_tensor = array_ops.boolean_mask(weight_tensor, is_id_valid)
234 math_ops.unsorted_segment_sum(weight_tensor, idx,
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc128 const TensorT* weight_tensor, in SymmetricPerChannelBiasQuantize() argument
142 if (tensor->shape[0] != weight_tensor->shape[channel_dim_index]) { in SymmetricPerChannelBiasQuantize()
145 tensor->shape[0], weight_tensor->shape[channel_dim_index]); in SymmetricPerChannelBiasQuantize()
154 TF_LITE_ENSURE(error_reporter, weight_tensor->quantization); in SymmetricPerChannelBiasQuantize()
155 const std::vector<float>& weight_scales = weight_tensor->quantization->scale; in SymmetricPerChannelBiasQuantize()
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead.py656 weight_tensor = _weight_tensor(features, weight_column_name)
657 loss, weighted_average_loss = loss_fn(labels, logits, weight_tensor)
669 batch_size, loss_fn, weight_tensor)
671 weighted_average_loss, predictions, labels, weight_tensor)
1787 weight_tensor = math_ops.cast(features[weight_column_name], dtypes.float32)
1788 shape = weight_tensor.get_shape()
1796 sparse_ops.sparse_reshape(weight_tensor, (-1, 1))
1797 if isinstance(weight_tensor, sparse_tensor.SparseTensor) else
1798 array_ops.reshape(weight_tensor, (-1, 1)))
1799 return weight_tensor
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column_test.py700 self.assertIsNone(id_weight_pair.weight_tensor)
739 self.assertIsNone(id_weight_pair.weight_tensor)
792 self.assertIsNone(id_weight_pair.weight_tensor)
817 self.assertIsNone(id_weight_pair.weight_tensor)
857 self.assertIsNone(id_weight_pair.weight_tensor)
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py225 weight_tensor = tensor[1]
234 indices=indices, values=weight_tensor.values, dense_shape=shape)