Home
last modified time | relevance | path

Searched refs:UseMultipleStreams (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.h54 bool UseMultipleStreams() const { in UseMultipleStreams() function
55 return xla_device_metadata_ && xla_device_metadata_->UseMultipleStreams(); in UseMultipleStreams()
Dxla_ops.cc354 platform_info_.UseMultipleStreams()); in Compute()
523 /*use_multiple_streams=*/platform_info_.UseMultipleStreams()); in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc153 if (UseMultipleStreams() && in CopyCPUTensorToDevice()
164 if (UseMultipleStreams()) { in CopyCPUTensorToDevice()
182 if (UseMultipleStreams()) { in CopyCPUTensorToDevice()
Dxla_device_context.h90 bool UseMultipleStreams() const { return stream_ != host_to_device_stream_; } in UseMultipleStreams() function
Dxla_device.h77 bool UseMultipleStreams() const { return use_multiple_streams_; } in UseMultipleStreams() function
Dxla_compile_on_demand_op.cc62 /*use_multiple_streams=*/metadata.UseMultipleStreams()); in Run()