Searched refs:UseMultipleStreams (Results 1 – 6 of 6) sorted by relevance
54 bool UseMultipleStreams() const { in UseMultipleStreams() function55 return xla_device_metadata_ && xla_device_metadata_->UseMultipleStreams(); in UseMultipleStreams()
354 platform_info_.UseMultipleStreams()); in Compute()523 /*use_multiple_streams=*/platform_info_.UseMultipleStreams()); in Compute()
153 if (UseMultipleStreams() && in CopyCPUTensorToDevice()164 if (UseMultipleStreams()) { in CopyCPUTensorToDevice()182 if (UseMultipleStreams()) { in CopyCPUTensorToDevice()
90 bool UseMultipleStreams() const { return stream_ != host_to_device_stream_; } in UseMultipleStreams() function
77 bool UseMultipleStreams() const { return use_multiple_streams_; } in UseMultipleStreams() function
62 /*use_multiple_streams=*/metadata.UseMultipleStreams()); in Run()