Searched refs:set_stream (Results 1 – 13 of 13) sorted by relevance
38 ExecutableRunOptions& ExecutableRunOptions::set_stream( in set_stream() function in xla::ExecutableRunOptions
56 ExecutableRunOptions& set_stream(stream_executor::Stream* stream);
313 def set_stream(file_object): function315 return set_stream
663 DefaultExecutableRunOptions().set_stream(&stream)); in XLA_TEST_F()687 DefaultExecutableRunOptions().set_stream(&wrong_stream)); in XLA_TEST_F()726 DefaultExecutableRunOptions().set_stream(&stream)); in XLA_TEST_F()936 run_options.set_allocator(&allocator).set_stream(stream.get()); in BM_LocalClientOverhead()
171 exec_run_options.set_stream(stream_ptr.get()); in ExecuteAndFetchProfile()
904 options.set_allocator(&allocator).set_stream(&stream); in BM_ParallelFusion()
143 run_options.set_stream(stream_); in CompareEqualImpl()
361 run_options.set_stream(stream); in Compute()535 run_options.set_stream(stream); in Compute()
76 run_options.set_stream(stream); in Run()
208 run_options.set_stream(stream); in DoWork()
157 run_options.set_stream(stream.get()); in Run()
449 options.set_stream(streams.back().get()); in ExecuteParallelAndRegisterResult()543 options.set_stream(stream.get()); in ExecuteAndRegisterResult()
446 run_options.set_stream(stream); in GetServiceRunOptionsForDevice()