Broadcast operator gradient performance test

gistfile1.txt

import TensorFlow
import Dispatch

func foo(x: Tensor<Float>, y: Tensor<Float>) -> Tensor<Float> {
  return (x + x + x * y * y * y).sum()
}

func time(_ body: () -> Void) {
  let divisor: Float = 1_000_000_000
  let start = Float(DispatchTime.now().uptimeNanoseconds) / divisor
  print("Start: \(start)s")
  body()
  let end = Float(DispatchTime.now().uptimeNanoseconds) / divisor
  print("End: \(end)s")
  let elapsed = end - start
  print("Elapsed: \(elapsed)s")
}

let x = Tensor<Float>(zeros: [10, 1, 10, 1, 1])
let y = Tensor<Float>(zeros: [1, 10, 1, 10, 1])

time {
  for i in 0..<10000 {
    _ = gradient(at: x, y, in: foo)
  }
}