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

Method MakeShapeFromShapeProto

tensorflow/core/framework/shape_inference.cc:955–961  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

953}
954
955Status InferenceContext::MakeShapeFromShapeProto(const TensorShapeProto& proto,
956 ShapeHandle* out) {
957 *out = nullptr;
958 TF_RETURN_IF_ERROR(PartialTensorShape::IsValidShape(proto));
959 PartialTensorShape partial_shape(proto);
960 return MakeShapeFromPartialTensorShape(partial_shape, out);
961}
962
963Status InferenceContext::GetScalarFromTensor(const Tensor* t, int64* val) {
964 // Caller must ensure that <t> is not NULL.

Callers 10

function_ops.ccFile · 0.80
TEST_FFunction · 0.80
UpdateFunctionMethod · 0.80
ValidateShapeMethod · 0.80
SetHandleShapeAndTypeFunction · 0.80

Calls

no outgoing calls

Tested by 1

TEST_FFunction · 0.64