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.cc40 GetStatTypeStr(StatType::kDevCapMemorySize)), in SetDeviceCaps()
70 case StatType::kDevCapMemorySize: in GetDeviceCaps()
Dxplane_schema.cc266 {"memory_size", kDevCapMemorySize}, in GetStatTypeMap()
Dxplane_schema.h258 kDevCapMemorySize, enumerator
/external/tensorflow/tensorflow/core/profiler/backends/gpu/
Ddevice_tracer_test.cc312 EXPECT_TRUE(plane.GetStat(kDevCapMemorySize).has_value()); in TEST_F()
Dcupti_collector.cc388 GetStatTypeStr(StatType::kDevCapMemorySize)), in GetDeviceCapabilities()
Ddevice_tracer_rocm.cc577 GetStatTypeStr(StatType::kDevCapMemorySize)), in GetDeviceCapabilities()