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_test.cc56 TF_EXPECT_OK(EnsureMemoryTypes(DEVICE_GPU, "/device:GPU:0", g)); in TEST()
65 TF_EXPECT_OK(EnsureMemoryTypes(DEVICE_SYCL, "/device:SYCL:0", g)); in TEST()
Dmemory_types.h38 Status EnsureMemoryTypes(const DeviceType& device_type,
Dmemory_types.cc150 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes() function
Ddirect_session.cc1297 TF_RETURN_IF_ERROR(EnsureMemoryTypes(DeviceType(device->device_type()), in CreateExecutors()
Dfunction.cc876 TF_RETURN_IF_ERROR(EnsureMemoryTypes(DeviceType(device()->device_type()), in CreateItem()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc269 EnsureMemoryTypes(DeviceType(unit->device->device_type()), in InitItem()