| 72 | } |
| 73 | |
| 74 | XlaResource* XlaContext::AddResource(std::unique_ptr<XlaResource> resource) { |
| 75 | resources_.push_back(std::move(resource)); |
| 76 | return resources_.back().get(); |
| 77 | } |
| 78 | |
| 79 | const xla::XlaComputation* XlaContext::GetOrCreateMax(const DataType type) { |
| 80 | return LookupOrCreate(type, &max_func_, [type] { |