Home
last modified time | relevance | path

Searched refs:use_multiple_streams_ (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.h80 bool UseMultipleStreams() const { return use_multiple_streams_; } in UseMultipleStreams()
89 const bool use_multiple_streams_; variable
246 const bool use_multiple_streams_; variable
H A Dxla_launch_util.cc218 use_multiple_streams_(use_multiple_streams), in XlaComputationLaunchContext()
220 if (use_multiple_streams_) { in XlaComputationLaunchContext()
287 if (use_multiple_streams_) { in PopulateInputs()
489 if (use_multiple_streams_) { in PopulateOutputs()
562 allocate_xla_tensors_, stream, use_multiple_streams_, in PopulateOutputs()
602 use_multiple_streams_, definition_event)); in PopulateOutputs()
H A Dxla_device.cc155 use_multiple_streams_(use_multiple_streams) {} in Metadata()
217 use_multiple_streams_(options.use_multiple_streams), in XlaDevice()
334 if (use_multiple_streams_) { in GetDeviceContextLocked()
H A Dxla_launch_util.h193 bool use_multiple_streams_; variable