(binding: TFJSBinding, packageName: string)
| 40 | private tensorMap: tf.DataStorage<TensorData>; |
| 41 | |
| 42 | constructor(binding: TFJSBinding, packageName: string) { |
| 43 | super(); |
| 44 | this.binding = binding; |
| 45 | this.isGPUPackage = packageName === '@tensorflow/tfjs-node-gpu'; |
| 46 | this.isUsingGpuDevice = this.binding.isUsingGpuDevice(); |
| 47 | this.tensorMap = new tf.DataStorage<TensorData>(this, tf.engine()); |
| 48 | } |
| 49 | |
| 50 | getDTypeInteger(dtype: DataType): number { |
| 51 | switch (dtype) { |
nothing calls this directly
no test coverage detected