Searched refs:ModelWeights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_internal.h | 123 class ModelWeights; variable
134 const int num_loss_partitions, const ModelWeights& model_weights,
188 friend class ModelWeights; variable
278 class ModelWeights {
280 ModelWeights() {} in ModelWeights() function
310 TF_DISALLOW_COPY_AND_ASSIGN(ModelWeights);
330 const ModelWeights& model_weights,
342 Status Initialize(OpKernelContext* const context, const ModelWeights& weights,
353 int num_sparse_features, const ModelWeights& weights,
364 int num_dense_features, const ModelWeights& weights,
|
| D | sdca_internal.cc | 79 void ModelWeights::UpdateDeltaWeights( in UpdateDeltaWeights()
95 Status ModelWeights::Initialize(OpKernelContext* const context) { in Initialize()
155 const int num_loss_partitions, const ModelWeights& model_weights, in ComputeWxAndWeightedExampleNorm()
237 const ModelWeights& model_weights, in SampleAdaptiveProbabilities()
321 const ModelWeights& weights, in Initialize()
388 const int num_sparse_features, const ModelWeights& weights, in CreateSparseFeatureRepresentation()
468 const int num_dense_features, const ModelWeights& weights, in CreateDenseFeatureRepresentation()
|
| D | sdca_ops.cc | 65 using sdca::ModelWeights;
132 ModelWeights model_weights; in DoCompute()
|