MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / UpdateDeltaWeights

Method UpdateDeltaWeights

tensorflow/core/kernels/sdca_internal.cc:79–93  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

77}
78
79void ModelWeights::UpdateDeltaWeights(
80 const Eigen::ThreadPoolDevice& device, const Example& example,
81 const std::vector<double>& normalized_bounded_dual_delta) {
82 // Sparse weights.
83 for (size_t j = 0; j < sparse_weights_.size(); ++j) {
84 sparse_weights_[j].UpdateSparseDeltaWeights(
85 device, example.sparse_features_[j], normalized_bounded_dual_delta);
86 }
87
88 // Dense weights.
89 for (size_t j = 0; j < dense_weights_.size(); ++j) {
90 dense_weights_[j].UpdateDenseDeltaWeights(
91 device, *example.dense_vectors_[j], normalized_bounded_dual_delta);
92 }
93}
94
95Status ModelWeights::Initialize(OpKernelContext* const context) {
96 OpInputList sparse_indices_inputs;

Callers 1

DoComputeFunction · 0.80

Calls 3

sizeMethod · 0.45

Tested by

no test coverage detected