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

Function MakeRead

tensorflow/compiler/jit/mark_for_compilation_pass_test.cc:648–658  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

646
647namespace {
648Node* MakeRead(const Scope& scope, const string& id,
649 Node** var_handle_op = nullptr) {
650 Output var_handle =
651 ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({}));
652 Output read =
653 ops::ReadVariableOp(scope.WithOpName("Read" + id), var_handle, DT_FLOAT);
654 if (var_handle_op) {
655 *var_handle_op = var_handle.node();
656 }
657 return read.node();
658}
659
660Node* MakeWrite(const Scope& scope, const string& id) {
661 Output var_handle =

Callers 1

TESTFunction · 0.70

Calls 5

WithOpNameMethod · 0.80
VarHandleOpClass · 0.50
TensorShapeClass · 0.50
ReadVariableOpClass · 0.50
nodeMethod · 0.45

Tested by

no test coverage detected