| 181 | } |
| 182 | |
| 183 | Status Add::Compile(const CreationContext& creation_context) { |
| 184 | const auto code = |
| 185 | GetElementWiseCode(definition_.src_tensors[0], definition_.dst_tensors[0], |
| 186 | definition_.precision, linked_operations_); |
| 187 | return creation_context.cache->GetOrCreateCLKernel( |
| 188 | code, "main_function", *creation_context.context, |
| 189 | *creation_context.device, &kernel_); |
| 190 | } |
| 191 | |
| 192 | Add CreateAdd(const OperationDef& definition, const std::vector<int>& channels, |
| 193 | int dst_channels) { |