Searched refs:allocate_xla_tensors_ (Results 1 – 2 of 2) sorted by relevance
194 allocate_xla_tensors_(allocate_xla_tensors), in XlaComputationLaunchContext()198 CHECK(allocate_xla_tensors_) << "To use multiple streams correctly we must " in XlaComputationLaunchContext()509 if (allocate_xla_tensors_) { in PopulateOutputs()556 if (allocate_xla_tensors_) { in PopulateOutputs()
184 bool allocate_xla_tensors_; variable