HomeSort by relevance Sort by last modified time
    Searched full:shape_pb (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/profiler/internal/
tfprof_node.cc 265 std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb) {
267 if (shape_pb.dim_size() == 0 && !shape_pb.unknown_rank()) {
271 for (const auto& d : shape_pb.dim()) {
279 TensorShapeProto shape_pb; local
281 shape_pb.set_unknown_rank(true);
282 return shape_pb;
285 shape_pb.add_dim()->set_size(s);
287 return shape_pb;
tfprof_node.h 39 std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb);

Completed in 134 milliseconds