Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dprofiler_test.py46 profile_pb = trace_events_pb2.Trace()
47 profile_pb.ParseFromString(profile_result)
48 devices = frozenset(device.name for device in profile_pb.devices.values())
53 events = frozenset(event.name for event in profile_pb.trace_events)
61 profile_pb = trace_events_pb2.Trace()
62 profile_result = profile_pb.SerializeToString()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_code.cc287 pprof::Profile profile_pb; in WritePprofProfile() local
288 Build(&profile_pb); in WritePprofProfile()
302 s = zlib_output_buffer->Append(profile_pb.SerializeAsString()); in WritePprofProfile()
320 void Build(pprof::Profile* profile_pb) { in Build() argument
322 auto sample_type = profile_pb->mutable_sample_type()->Add(); in Build()
328 sample_type = profile_pb->mutable_sample_type()->Add(); in Build()
333 profile_pb->mutable_comment()->Add(string_table_.GetIndex( in Build()
336 profile_pb->mutable_comment()->Add( in Build()
339 profile_pb->mutable_comment()->Add( in Build()
344 profile_pb->mutable_comment()->Add( in Build()
[all …]
/external/tensorflow/tensorflow/python/profiler/
Dmodel_analyzer_test.py442 profile_pb = profile_pb2.Profile()
445 profile_pb.ParseFromString(gzipf.read())
447 self.assertGreater(len(profile_pb.sample), 10)
448 self.assertGreater(len(profile_pb.location), 10)
449 self.assertGreater(len(profile_pb.function), 10)
450 self.assertGreater(len(profile_pb.string_table), 30)
454 for s in profile_pb.string_table: