Searched refs:default_stream_executor (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 79 backend().default_stream_executor()); in TEST_F()
168 backend().default_stream_executor()); in TEST_F()
239 backend().default_stream_executor()); in TEST_F()
321 backend().default_stream_executor()); in TEST_F()
358 backend().default_stream_executor()); in TEST_F()
396 backend().default_stream_executor()); in TEST_F()
|
| D | cudnn_fused_conv_rewriter_test.cc | 36 backend().default_stream_executor(), in GetOptimizedHlo()
299 backend().default_stream_executor(), in TEST_F()
|
| D | buffer_comparator_test.cc | 31 stream_exec_(backend_->default_stream_executor()), in BufferComparatorTest()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 79 backend_->default_stream_executor(), in TestCompilerHooks()
101 executors.push_back({backend_->default_stream_executor()}); in TestMultiModuleCompilation()
102 executors.push_back({backend_->default_stream_executor()}); in TestMultiModuleCompilation()
|
| D | codegen_test_base.cc | 24 backend().default_stream_executor(), in CompileToExecutable()
27 backend().default_stream_executor(), in CompileToExecutable()
|
| D | xla_hlo_profile_test.cc | 137 se::StreamExecutor* executor = backend->default_stream_executor(); in ExecuteAndFetchProfile()
|
| D | tuple_test.cc | 576 backend().default_stream_executor(), expected.shape(), literal)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 104 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralToDevice()
135 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralFromDevice()
206 se::Stream stream(backend().default_stream_executor()); in ExecuteWithDeviceBuffers()
243 se::Stream stream(backend().default_stream_executor()); in ExecuteWithDeviceBuffers()
433 {{backend().default_stream_executor()}}, in CreateExecutable()
438 backend().default_stream_executor(), in CreateExecutable()
|
| D | hlo_execution_profile_test.cc | 69 backend().default_stream_executor()->GetDeviceDescription()), in TEST_F()
|
| D | layout_assignment_test.cc | 684 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
691 backend().default_stream_executor(), in TEST_F()
943 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
975 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1008 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1041 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1065 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
|
| D | backend.h | 116 se::StreamExecutor* default_stream_executor() const { in default_stream_executor() function
|
| D | backend.cc | 175 return default_stream_executor()->device_ordinal(); in default_device_ordinal()
|
| D | service.cc | 846 execute_backend_->default_stream_executor(), in Compile()
893 execute_backend_->default_stream_executor())); in Execute()
|