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

Function CreateAssignShapeFn

tensorflow/core/ops/kv_variable_ops.cc:60–69  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

58}
59
60Status CreateAssignShapeFn(InferenceContext* c) {
61 ShapeAndType handle_shape_and_type;
62 TF_RETURN_IF_ERROR(ValidateVariableResourceHandle(c, 0, &handle_shape_and_type));
63
64 ShapeHandle value_shape = c->input(1);
65 ShapeHandle unused;
66 TF_RETURN_IF_ERROR(
67 c->Merge(handle_shape_and_type.shape, value_shape, &unused));
68 return Status::OK();
69}
70
71} // namespace
72

Callers

nothing calls this directly

Calls 3

inputMethod · 0.45
MergeMethod · 0.45

Tested by

no test coverage detected