Searched refs:set_stream (Results 1 – 14 of 14) sorted by relevance
52 ExecutableRunOptions& ExecutableRunOptions::set_stream( in set_stream() function in xla::ExecutableRunOptions
86 ExecutableRunOptions& set_stream(stream_executor::Stream* stream);
313 def set_stream(file_object): function315 return set_stream
666 DefaultExecutableRunOptions().set_stream(&stream)); in XLA_TEST_F()690 DefaultExecutableRunOptions().set_stream(&wrong_stream)); in XLA_TEST_F()729 DefaultExecutableRunOptions().set_stream(&stream)); in XLA_TEST_F()963 run_options.set_allocator(&allocator).set_stream(stream.get()); in BM_LocalClientOverhead()
77 run_options.set_stream(&stream); in RunAndCheck()
171 exec_run_options.set_stream(stream_ptr.get()); in ExecuteAndFetchProfile()
904 options.set_allocator(&allocator).set_stream(&stream); in BM_ParallelFusion()
77 run_options.set_stream(stream); in Run()
383 run_options.set_stream(stream); in Compute()591 run_options.set_stream(stream); in Compute()
155 run_options.set_stream(stream.get()); in RunHelper()
154 run_options.set_stream(stream); in RunExecutable()
465 options.set_stream(streams.back().get()); in ExecuteParallelAndRegisterResult()561 options.set_stream(stream.get()); in ExecuteAndRegisterResult()
451 run_options.set_stream(stream); in GetServiceRunOptionsForDevice()
773 options.set_stream(device_state->compute_stream()); in ExecuteHelper()