Searched refs:set_stream (Results 1 - 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
H A Dexecutable_run_options.cc38 ExecutableRunOptions& ExecutableRunOptions::set_stream( function in class:xla::ExecutableRunOptions
H A Dexecutable_run_options.h64 ExecutableRunOptions& set_stream(perftools::gputools::Stream* stream);
/external/autotest/client/common_lib/
H A Dlogging_manager.py313 def set_stream(file_object): function in function:LoggingManager._sys_stream_setter
315 return set_stream
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_runner.cc134 run_options.set_stream(&stream);
H A Dservice.cc538 options.set_stream(streams.back().get());
654 options.set_stream(stream.get());
966 options.set_stream(stream.get());
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dlocal_client.cc153 run_options.set_stream(stream.get());
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_execute_test.cc689 DefaultExecutableRunOptions().set_stream(&stream));
714 DefaultExecutableRunOptions().set_stream(&wrong_stream));
753 DefaultExecutableRunOptions().set_stream(&stream));
936 run_options.set_allocator(&allocator).set_stream(&stream);
H A Dxla_hlo_profile_test.cc158 exec_run_options.set_stream(stream_ptr.get());
H A Dfusion_test.cc835 options.set_allocator(&allocator).set_stream(&stream);
/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_launch_op.cc313 run_options.set_stream(stream);

Completed in 258 milliseconds