/external/tensorflow/tensorflow/compiler/xla/tests/ |
codegen_test_base.cc | 24 backend().default_stream_executor(), 27 backend().default_stream_executor(),
|
llvm_compiler_test.cc | 79 backend_->default_stream_executor(), 101 executors.push_back({backend_->default_stream_executor()}); 102 executors.push_back({backend_->default_stream_executor()});
|
xla_hlo_profile_test.cc | 137 se::StreamExecutor* executor = backend->default_stream_executor();
|
tuple_test.cc | 576 backend().default_stream_executor(), expected.shape(), literal));
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
cudnn_fused_conv_rewriter_test.cc | 36 backend().default_stream_executor(), 299 backend().default_stream_executor(),
|
gpu_layout_assignment_test.cc | 79 backend().default_stream_executor()); 168 backend().default_stream_executor()); 239 backend().default_stream_executor()); 321 backend().default_stream_executor()); 358 backend().default_stream_executor()); 396 backend().default_stream_executor());
|
buffer_comparator_test.cc | 31 stream_exec_(backend_->default_stream_executor()),
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
hlo_execution_profile_test.cc | 69 backend().default_stream_executor()->GetDeviceDescription()),
|
hlo_runner.cc | 104 auto stream, backend().BorrowStream(backend().default_stream_executor())); 135 auto stream, backend().BorrowStream(backend().default_stream_executor())); 206 se::Stream stream(backend().default_stream_executor()); 243 se::Stream stream(backend().default_stream_executor()); 433 {{backend().default_stream_executor()}}, 438 backend().default_stream_executor(),
|
backend.h | 116 se::StreamExecutor* default_stream_executor() const { function in class:xla::Backend
|
backend.cc | 175 return default_stream_executor()->device_ordinal();
|
layout_assignment_test.cc | 684 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), 691 backend().default_stream_executor(), [all...] |
service.cc | [all...] |