Searched refs:model_weights (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 124 ModelWeights model_weights; in DoCompute() local
125 OP_REQUIRES_OK(context, model_weights.Initialize(context)); in DoCompute()
130 examples.Initialize(context, model_weights, options.num_sparse_features, in DoCompute()
154 model_weights, example_state_data, options.loss_updater, in DoCompute()
186 options.num_loss_partitions, model_weights, in DoCompute()
197 model_weights.UpdateDeltaWeights( in DoCompute()
|
| D | sdca_internal.cc | 149 const int num_loss_partitions, const ModelWeights& model_weights, in ComputeWxAndWeightedExampleNorm() argument
161 model_weights.sparse_weights()[j]; in ComputeWxAndWeightedExampleNorm()
185 model_weights.dense_weights()[j]; in ComputeWxAndWeightedExampleNorm()
231 const ModelWeights& model_weights, in SampleAdaptativeProbabilities() argument
248 model_weights, regularization, in SampleAdaptativeProbabilities()
|
| D | sdca_internal.h | 136 const int num_loss_partitions, const ModelWeights& model_weights,
335 const ModelWeights& model_weights,
|
| /external/tensorflow/tensorflow/python/estimator/canned/ |
| D | dnn_testing_utils.py | 101 model_weights = {}
105 model_weights[HIDDEN_WEIGHTS_NAME_PATTERN % i] = weights[i]
106 model_weights[HIDDEN_BIASES_NAME_PATTERN % i] = biases[i]
109 model_weights[LOGITS_WEIGHTS_NAME] = weights[-1]
110 model_weights[LOGITS_BIASES_NAME] = biases[-1]
114 for k, v in six.iteritems(model_weights):
|