Searched refs:use_multiple_streams (Results 1 – 7 of 7) sorted by relevance
64 PaddedShapeFn padded_shape_fn, bool use_multiple_streams);120 bool use_multiple_streams = false; member
149 PaddedShapeFn padded_shape_fn, bool use_multiple_streams) in Metadata() argument155 use_multiple_streams_(use_multiple_streams) {} in Metadata()206 options.use_multiple_streams), in XlaDevice()212 use_multiple_streams_(options.use_multiple_streams), in XlaDevice()
89 options.use_multiple_streams = false; in CreateDevices()
191 int device_ordinal, bool allocate_xla_tensors, bool use_multiple_streams) in XlaComputationLaunchContext() argument195 use_multiple_streams_(use_multiple_streams), in XlaComputationLaunchContext()332 se::Stream* stream, bool use_multiple_streams, in PopulateXlaTensor() argument337 if (use_multiple_streams) { in PopulateXlaTensor()
138 bool use_multiple_streams);
155 options.use_multiple_streams = true; in CreateDevices()
373 options.use_multiple_streams = true; in CreateDevices()