Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc245 device_fns_(device_fns), in CStreamExecutor()
620 if (device_fns_->get_numa_node != nullptr) { in CreateDeviceDescription()
621 int32_t numa_node = device_fns_->get_numa_node(&device_); in CreateDeviceDescription()
627 if (device_fns_->get_memory_bandwidth != nullptr) { in CreateDeviceDescription()
628 int64_t memory_bandwidth = device_fns_->get_memory_bandwidth(&device_); in CreateDeviceDescription()
663 SP_DeviceFns* device_fns_; member in stream_executor::__anon00950a620111::CStreamExecutor
683 device_fns_(std::move(device_fns)), in CPlatform()
690 platform_fns_.destroy_device_fns(&platform_, &device_fns_); in ~CPlatform()
739 std::move(device), &device_fns_, &stream_executor_, &platform_, in GetUncachedExecutor()
Dstream_executor_test.cc319 PopulateDefaultDeviceFns(&device_fns_); in SetUp()
329 device_fns_, se_, timer_fns_); in GetExecutor()
338 SP_DeviceFns device_fns_; member in stream_executor::__anon40285f810111::StreamExecutorTest
942 device_fns_.get_numa_node = [](const SP_Device* device) { return 123; }; in TEST_F()
943 device_fns_.get_memory_bandwidth = [](const SP_Device* device) -> int64_t { in TEST_F()
946 device_fns_.get_gflops = [](const SP_Device* device) -> double { return 32; }; in TEST_F()
969 device_fns_.get_memory_bandwidth = [](const SP_Device* device) -> int64_t { in TEST_F()
972 device_fns_.get_gflops = [](const SP_Device* device) -> double { return 32; }; in TEST_F()
Dstream_executor_internal.h86 SP_DeviceFns device_fns_; variable