Home
last modified time | relevance | path

Searched defs:use_multiple_streams (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc214 int device_ordinal, bool allocate_xla_tensors, bool use_multiple_streams) in XlaComputationLaunchContext()
328 bool use_multiple_streams, std::shared_ptr<se::Event> definition_event) { in GetOrCreateTensorForOutput()
H A Dxla_device.h124 bool use_multiple_streams = false; member
H A Dxla_device.cc149 PaddedShapeFn padded_shape_fn, bool use_multiple_streams) in Metadata()