loc, operand_types, act, scale, mean, var, grad, op.getEpsilon(),
          feature_dim);

      x_backprop = training_op.getResult(0);

      scale_backprop = training_op.getResult(1);

      offset_backprop = training_op.getResult(2);
    } else {  // inference
      SmallVector<int64_t, 4> non_feature_dims;
      for (int64_t i = 0; i < act_type.getRank(); ++i) {

FailOnTimeout$Builder builder(); public void FailOnTimeout(org.junit.runners.model.Statement, long); private void FailOnTimeout(FailOnTimeout$Builder, org.junit.runners.model.Statement); public void evaluate() throws Throwable; private Throwable getResult(java.util.concurrent.FutureTask, Thread); private Exception createTimeoutExcepti(Thread); private StackTraceElement[] getStackTrace(Thread); private Thread getStuckThread(Thread); private java.util.List getThreadsInGroup(ThreadGroup); private long...