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

Function ReadVariableShapeFn

tensorflow/core/ops/kv_variable_ops.cc:53–58  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

51}
52
53Status ReadVariableShapeFn(InferenceContext* c) {
54 ShapeAndType shape_and_type;
55 TF_RETURN_IF_ERROR(ValidateVariableResourceHandle(c, 0, &shape_and_type));
56 c->set_output(0, shape_and_type.shape);
57 return Status::OK();
58}
59
60Status CreateAssignShapeFn(InferenceContext* c) {
61 ShapeAndType handle_shape_and_type;

Callers

nothing calls this directly

Calls 2

set_outputMethod · 0.45

Tested by

no test coverage detected