Function assertNotComplex

tfjs-backend-cpu/src/cpu_util.ts:20–33

(
    tensor: TensorInfo|TensorInfo[], opName: string)

import {TensorInfo, util} from '@tensorflow/tfjs-core';

export function assertNotComplex(
    tensor: TensorInfo|TensorInfo[], opName: string): void {
  if (!Array.isArray(tensor)) {
    tensor = [tensor];
  }
  tensor.forEach(t => {
    if (t != null) {
      util.assert(
          t.dtype !== 'complex64',
          () => `${
              opName} does not support complex64 tensors in the CPU backend.`);
    }
  });
}