Searched defs:use_multiple_streams (Results 1 – 3 of 3) sorted by relevance
120 bool use_multiple_streams = false; member
191 int device_ordinal, bool allocate_xla_tensors, bool use_multiple_streams) in XlaComputationLaunchContext()332 se::Stream* stream, bool use_multiple_streams, in PopulateXlaTensor()
149 PaddedShapeFn padded_shape_fn, bool use_multiple_streams) in Metadata()