Searched defs:weight_tensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | custom_loss_head.py | 53 def loss_wrapper(labels, logits, weight_tensor): argument
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | quantize.py | 592 def __init__(self, layer_op, weight_tensor, activation_op, bypass_op, argument
606 def weight_tensor(self): member in _LayerMatch
|
| D | fold_batch_norms.py | 446 def _CloneWithNewOperands(layer_op, input_tensor, weight_tensor, argument
1002 weight_tensor, gamma_tensor, beta_tensor, mean_tensor, argument
1039 def weight_tensor(self): member in _BatchNormMatch
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | subgraph_quantizer.cc | 128 const TensorT* weight_tensor, in SymmetricPerChannelBiasQuantize()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column.py | 419 def weight_tensor(self, input_tensor): member in _SparseColumn
815 def weight_tensor(self, input_tensor): member in _WeightedSparseColumn
838 def _do_transform(self, id_tensor, weight_tensor): argument
2335 def weight_tensor(self, input_tensor): member in _CrossedColumn
|
| D | target_column.py | 196 def _weighted_loss(self, loss, weight_tensor): argument
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 3908 def _transform_weight_tensor(self, weight_tensor): argument
|