Searched refs:ShapeProtoToVec (Results 1 – 3 of 3) sorted by relevance
38 std::vector<int64_t> ShapeProtoToVec(const TensorShapeProto& shape_pb);331 shape_ = ShapeProtoToVec(attr.second.shape()); in TFGraphNode()338 output_shapes_[i] = ShapeProtoToVec(attr.second.list().shape(i)); in TFGraphNode()
257 std::vector<int64_t> ShapeProtoToVec(const TensorShapeProto& shape_pb) { in ShapeProtoToVec() function
172 input_shapes[inp.first] = ShapeProtoToVec(inp.second); in FormatInputShapes()