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

Method CgmodeCompileOp

tensorflow/compiler/jit/kernels/cuda_graph_mode_ops.cc:151–156  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

149} // namespace
150
151CgmodeCompileOp::CgmodeCompileOp(OpKernelConstruction* ctx)
152 : OpKernel(ctx), function_(FunctionAttr(ctx)) {
153 env_ = ctx->env();
154 flib_ = ctx->function_library();
155 is_compiled_ = false;
156}
157
158Status CgmodeCompileOp::Compile(OpKernelContext* ctx, tstring& compiled_key) {
159 auto gpu_device = dynamic_cast<BaseGPUDevice*>(ctx->device());

Callers

nothing calls this directly

Calls 3

FunctionAttrFunction · 0.70
envMethod · 0.45
function_libraryMethod · 0.45

Tested by

no test coverage detected