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

Function InitialStack

tensorflow/c/eager/tape.h:538–548  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

536
537template <typename BackwardFunction, typename TapeTensor>
538std::vector<int64> InitialStack(
539 const OpTape<BackwardFunction, TapeTensor>& op_tape,
540 const std::unordered_map<int64, int64>& op_missing_tensor) {
541 std::vector<int64> result;
542 for (auto& op_entry : op_tape) {
543 if (op_missing_tensor.find(op_entry.first) == op_missing_tensor.end()) {
544 result.push_back(op_entry.first);
545 }
546 }
547 return result;
548}
549
550template <typename Gradient, typename BackwardFunction, typename TapeTensor>
551Status InitialGradients(

Callers 1

ComputeGradientMethod · 0.85

Calls 3

findMethod · 0.45
endMethod · 0.45
push_backMethod · 0.45

Tested by

no test coverage detected