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

Function ArgOp

tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc:479–482  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

477}
478
479Node* ArgOp(int index, DataType type, const GraphDefBuilder::Options& opts) {
480 return ops::SourceOp("_Arg",
481 opts.WithAttr("T", type).WithAttr("index", index));
482}
483
484Node* RetOp(int index, ops::NodeOut a, const GraphDefBuilder::Options& opts) {
485 if (opts.HaveError()) return nullptr;

Callers

nothing calls this directly

Calls 2

SourceOpFunction · 0.85
WithAttrMethod · 0.80

Tested by

no test coverage detected