Home
last modified time | relevance | path

Searched refs:kDevCapMemorySize (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/utils/
Ddevice_caps_utils.cc39 GetStatTypeStr(StatType::kDevCapMemorySize)), in SetDeviceCaps()
66 case StatType::kDevCapMemorySize: in GetDeviceCaps()
Dxplane_schema.h205 kDevCapMemorySize, enumerator
Dxplane_schema.cc213 {"memory_size", kDevCapMemorySize}, in GetStatTypeMap()
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_op_stats.cc66 case kDevCapMemorySize: in GetDeviceCapFromXPlane()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc283 EXPECT_TRUE(plane.GetStat(kDevCapMemorySize).has_value()); in TEST_F()
Dcupti_collector.cc529 GetStatTypeStr(StatType::kDevCapMemorySize)), in GetDeviceCapabilities()