Searched refs:platform_fns_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 220 platform_fns_(platform_fns), in CStreamExecutor()
226 platform_fns_->destroy_device(platform_, &device_); in ~CStreamExecutor()
651 SP_PlatformFns* platform_fns_; member in stream_executor::__anoncabb94000111::CStreamExecutor
666 platform_fns_(std::move(platform_fns)), in CPlatform()
675 platform_fns_.destroy_device_fns(&platform_, &device_fns_); in ~CPlatform()
676 platform_fns_.destroy_stream_executor(&platform_, &stream_executor_); in ~CPlatform()
677 platform_fns_.destroy_timer_fns(&platform_, &timer_fns_); in ~CPlatform()
679 destroy_platform_fns_(&platform_fns_); in ~CPlatform()
719 platform_fns_.create_device(&platform_, &device_params, c_status.get()); in GetUncachedExecutor()
725 platform_fns_.get_device_count(&platform_, &visible_device_count, in GetUncachedExecutor()
[all …]
|
| D | stream_executor_internal.h | 64 platform_fns_.get_device_count(&platform_, &visible_device_count, in VisibleDeviceCount()
95 SP_PlatformFns platform_fns_; variable
|
| D | stream_executor_test.cc | 142 test_util::PopulateDefaultPlatform(&platform_, &platform_fns_); in SetUp()
152 platform_, test_util::DestroyPlatform, platform_fns_, in GetExecutor()
161 SP_PlatformFns platform_fns_; member in stream_executor::__anon555b9bdf0111::StreamExecutorTest
758 platform_fns_.create_device = [](const SP_Platform* platform, in TEST_F()
785 platform_fns_.create_device = [](const SP_Platform* platform, in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | while_op_test.cc | 36 &platform_fns_); in SetUp()
44 SP_PlatformFns platform_fns_; member in tensorflow::__anon24af661f0111::WhileOpTest
166 std::move(platform_fns_), in TEST_F()
|