Searched refs:use_multiple_streams (Results 1 – 7 of 7) sorted by relevance
143 PaddedShapeFn padded_shape_fn, bool use_multiple_streams) in Metadata() argument149 use_multiple_streams_(use_multiple_streams) {} in Metadata()200 options.use_multiple_streams), in XlaDevice()204 use_multiple_streams_(options.use_multiple_streams), in XlaDevice()
64 PaddedShapeFn padded_shape_fn, bool use_multiple_streams);115 bool use_multiple_streams = false; member
63 options.use_multiple_streams = false; in CreateDevices()
65 options.use_multiple_streams = false; in CreateDevices()
100 options.use_multiple_streams = true; in CreateDevices()
147 bool use_multiple_streams);
196 bool allocate_xla_tensors, bool use_multiple_streams) in XlaComputationLaunchContext() argument200 use_multiple_streams_(use_multiple_streams) { in XlaComputationLaunchContext()