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

Function MakeWrite

tensorflow/compiler/jit/build_xla_ops_pass_test.cc:108–114  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

106}
107
108Node* MakeWrite(const Scope& scope, Output value_to_write, const string& id) {
109 Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT,
110 TensorShape({}));
111 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id),
112 var_handle, value_to_write);
113 return assign_op.operation.node();
114}
115
116Node* MakeWrite(const Scope& scope, const string& id) {
117 return MakeWrite(

Callers 1

TEST_FFunction · 0.70

Calls 5

WithOpNameMethod · 0.80
ConstFunction · 0.70
VarHandleOpClass · 0.50
TensorShapeClass · 0.50
nodeMethod · 0.45

Tested by

no test coverage detected