MCPcopy Create free account
hub / github.com/ROCm/AMDMIGraphX / parse_dims

Function parse_dims

src/tf/tf_parser.cpp:197–206  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

195}
196
197static std::vector<size_t> parse_dims(const tensorflow::TensorShapeProto& s)
198{
199 std::vector<size_t> dims;
200 auto input_dims = s.dim();
201 std::transform(input_dims.begin(),
202 input_dims.end(),
203 std::back_inserter(dims),
204 [](const tensorflow::TensorShapeProto_Dim& dim) { return dim.size(); });
205 return dims;
206}
207
208template <class T>
209static std::vector<T> get_data_vals(const google::protobuf::RepeatedField<T>& data,

Callers 2

parse_graphMethod · 0.85
parse_tensorMethod · 0.85

Calls 4

transformFunction · 0.50
beginMethod · 0.45
endMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected