| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| custom_loss_head.py | 53 def loss_wrapper(labels, logits, weight_tensor):
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)
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| quantize.py | 74 layer_match.weight_tensor.op, [layer_match.layer_op],
197 weight_tensor = match_result.get_tensor(weight_pattern)
198 if weight_tensor is None:
199 weight_tensor = match_result.get_tensor(folded_weight_pattern)
207 yield _LayerMatch(layer_op, weight_tensor, activation_op, bypass_op,
214 def __init__(self, layer_op, weight_tensor, activation_op, bypass_op,
217 self._weight_tensor = weight_tensor
227 def weight_tensor(self): member in class:_LayerMatch
|
| fold_batch_norms.py | 103 weights = match.weight_tensor
106 match.weight_tensor.get_shape().as_list()[2],
107 match.weight_tensor.get_shape().as_list()[3]
211 weight_tensor = match_result.get_tensor(weight_pattern)
264 weight_tensor=weight_tensor,
377 def _CloneWithNewOperands(layer_op, input_tensor, weight_tensor):
378 """Clones layer_op with input_tensor and weight_tensor as new inputs."""
383 weight_tensor,
392 weight_tensor,
852 def weight_tensor(self): member in class:_BatchNormMatch
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| target_column.py | 194 def _weighted_loss(self, loss, weight_tensor):
199 weight_tensor, shape=(-1,)))
226 weight_tensor = self.get_weight_tensor(features)
227 if weight_tensor is None:
229 loss_weighted = self._weighted_loss(loss_unweighted, weight_tensor)
251 weight_tensor = self.get_weight_tensor(features)
252 if weight_tensor is None:
254 loss_weighted = self._weighted_loss(loss_unweighted, weight_tensor)
256 math_ops.to_float(math_ops.reduce_sum(weight_tensor)),
|
| feature_column.py | 167 "weight_tensor",
181 "weight_tensor",
419 def weight_tensor(self, input_tensor): member in class:_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 class:_WeightedSparseColumn
833 weight_tensor=self.weight_tensor(input_tensor)
2331 def weight_tensor(self, input_tensor): member in class:_CrossedColumn
[all...] |
| feature_column_ops.py | 306 sparse_weights=embedding_lookup_arguments.weight_tensor,
319 assert arg.weight_tensor is None, (
[all...] |
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| sdca_optimizer.py | 191 weight_tensor = column.weight_tensor(transformed_tensor)
199 array_ops.reshape(weight_tensor.values, [-1])))
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column.py | [all...] |
| feature_column_test.py | 467 self.assertIsNone(id_weight_pair.weight_tensor)
482 self.assertIsNone(id_weight_pair.weight_tensor)
724 self.assertIsNone(id_weight_pair.weight_tensor)
746 self.assertIsNone(id_weight_pair.weight_tensor)
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| head.py | 638 weight_tensor = _weight_tensor(features, weight_column_name)
639 loss, weighted_average_loss = loss_fn(labels, logits, weight_tensor)
651 batch_size, loss_fn, weight_tensor)
653 weighted_average_loss, predictions, labels, weight_tensor)
[all...] |
