Searched refs:allocate_xla_tensors_ (Results 1 – 2 of 2) sorted by relevance
199 allocate_xla_tensors_(allocate_xla_tensors), in XlaComputationLaunchContext()202 CHECK(allocate_xla_tensors_) << "To use multiple streams correctly we must " in XlaComputationLaunchContext()355 if (allocate_xla_tensors_) { in PopulateOutputs()418 if (allocate_xla_tensors_) { in PopulateOutputs()
190 bool allocate_xla_tensors_; variable