| 81 | // Exported for tests. |
| 82 | export const esmImportProvider: ImportProvider = { |
| 83 | importCoreStr(forwardModeOnly: boolean) { |
| 84 | const importLines = [ |
| 85 | `import {registerKernel} from '@tensorflow/tfjs-core/dist/base';`, |
| 86 | `import '@tensorflow/tfjs-core/dist/base_side_effects';`, |
| 87 | `export * from '@tensorflow/tfjs-core/dist/base';` |
| 88 | ]; |
| 89 | |
| 90 | if (!forwardModeOnly) { |
| 91 | importLines.push( |
| 92 | `import {registerGradient} from '@tensorflow/tfjs-core/dist/base';`); |
| 93 | } |
| 94 | return importLines.join('\n'); |
| 95 | }, |
| 96 | |
| 97 | importConverterStr() { |
| 98 | return `export * from '@tensorflow/tfjs-converter';`; |