Home
last modified time | relevance | path

Searched refs:ShapeProtoToVec (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h39 std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb);
332 shape_ = ShapeProtoToVec(attr.second.shape()); in TFGraphNode()
339 output_shapes_[i] = ShapeProtoToVec(attr.second.list().shape(i)); in TFGraphNode()
Dtfprof_node.cc265 std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb) { in ShapeProtoToVec() function
Dtfprof_show_multi.cc173 input_shapes[inp.first] = ShapeProtoToVec(inp.second); in FormatInputShapes()