Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.h38 Status EnsureMemoryTypes(const DeviceType& device_type,
Dmemory_types_test.cc53 TF_EXPECT_OK(EnsureMemoryTypes(DEVICE_GPU, "/device:GPU:0", g)); in TEST()
Dmemory_types.cc154 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes() function
Dfunction.cc925 TF_RETURN_IF_ERROR(EnsureMemoryTypes(DeviceType(device()->device_type()), in CreateItem()
Ddirect_session.cc1411 TF_RETURN_IF_ERROR(EnsureMemoryTypes(DeviceType(device->device_type()), in CreateExecutors()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc283 EnsureMemoryTypes(DeviceType(unit->device->device_type()), in InitItem()