Function Softmax

tensorflow/core/kernels/mkl_softmax_op_test.cc:45–70

template <typename T>
static Graph* Softmax(const string& kind, const TensorShape& in_shape) {
  Graph* g = new Graph(OpRegistry::Global());
  DataType type = DataTypeToEnum<T>::v();

  const bool isDefault = (kind == "Default");
  string op_name = isDefault ? "Softmax" : "_MklSoftmax";

  // Create inputs
  Tensor input1(type, in_shape);
  input1.flat<T>().setRandom();

  Node* input_in0 = test::graph::Constant(g, input1);

  Node* not_mkl_shape =
      test::graph::Constant(g, GetMklMetaTensor(), "not_mkl");
  // Create NodeDef
  auto nodeBuilder = NodeBuilder(g->NewName("softmax"), op_name)
                  .Input(input_in0);

  isDefault ? nodeBuilder : nodeBuilder.Input(not_mkl_shape)
                                       .Attr("_kernel", "MklLayoutDependentOp");

  TF_CHECK_OK(nodeBuilder.Finalize(g, nullptr));

  return g;
}

#define BM_Softmax_Base(T, kind, name, in_shape, DEVICE, NTH)                \
  static void BM_Softmax_##T##_##kind##name##_##NTH(int iters) {             \