Searched refs:model_weights (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 131 ModelWeights model_weights; in DoCompute() local
132 OP_REQUIRES_OK(context, model_weights.Initialize(context)); in DoCompute()
137 examples.Initialize(context, model_weights, options.num_sparse_features, in DoCompute()
161 model_weights, example_state_data, options.loss_updater, in DoCompute()
195 options.num_loss_partitions, model_weights, in DoCompute()
206 model_weights.UpdateDeltaWeights( in DoCompute()
|
| D | sdca_internal.cc | 155 const int num_loss_partitions, const ModelWeights& model_weights, in ComputeWxAndWeightedExampleNorm() argument
167 model_weights.sparse_weights()[j]; in ComputeWxAndWeightedExampleNorm()
191 model_weights.dense_weights()[j]; in ComputeWxAndWeightedExampleNorm()
237 const ModelWeights& model_weights, in SampleAdaptiveProbabilities() argument
254 model_weights, regularization, in SampleAdaptiveProbabilities()
|
| D | sdca_internal.h | 134 const int num_loss_partitions, const ModelWeights& model_weights,
330 const ModelWeights& model_weights,
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 2237 model_weights = model.get_weights()
2239 model_weights = None
2255 if model_weights:
2256 cpu_model.set_weights(model_weights)
|