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

Method Compute

tensorflow/core/kernels/save_op.cc:33–35  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

31 explicit SaveOp(OpKernelConstruction* context) : OpKernel(context) {}
32
33 void Compute(OpKernelContext* context) override {
34 SaveTensors(context, &checkpoint::CreateTableTensorSliceBuilder, false);
35 }
36};
37
38REGISTER_KERNEL_BUILDER(Name("Save").Device(DEVICE_CPU), SaveOp);

Callers

nothing calls this directly

Calls 1

SaveTensorsFunction · 0.85

Tested by

no test coverage detected