Searched refs:model_weights (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 132 ModelWeights model_weights; in DoCompute() local
133 OP_REQUIRES_OK(context, model_weights.Initialize(context)); in DoCompute()
138 examples.Initialize(context, model_weights, options.num_sparse_features, in DoCompute()
162 model_weights, example_state_data, options.loss_updater, in DoCompute()
196 options.num_loss_partitions, model_weights, in DoCompute()
207 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/python/keras/distribute/ |
| D | distribute_strategy_test.py | 1542 model_weights = model_with_array_input.get_weights()
1554 model_with_dict_input.set_weights(model_weights)
|