Method PrepareAndExecuteNet

tensorflow/core/kernels/mkl_conv_ops.cc:1043–1082

TODO(intel-mkl): This function does not seem to be called. Remove it. Prepare and execute net - checks for input and output reorders.

  // TODO(intel-mkl): This function does not seem to be called. Remove it.
  // Prepare and execute net - checks for input and output reorders.
  void PrepareAndExecuteNet(const ConvFwdPd& conv_prim_desc,
                            MklDnnData<Tinput>* src,
                            MklDnnData<Tfilter>* filter,
                            MklDnnData<Tbias>* bias,
                            MklDnnData<Toutput>* output,
                            Tensor* filter_out_tensor) {
    DCHECK(filter_out_tensor);

    // Create reorders between user layout and OneDNN layout if it is needed and
    // add it to the net before convolution. No need to check for output
    // reorder as we propagate output layout to the next layer.
    src->CheckReorderToOpMem(
        MEMORY_PD_WITHOUT_DATA(conv_prim_desc.PRIMITIVE_DESC_SRC, cpu_engine_));

    // Rather than re-ordering to a temp buffer, reorder directly to the
    // filter output tensor
    filter->CheckReorderToOpMem(conv_prim_desc.PRIMITIVE_DESC_WEIGHTS,
                                filter->GetTensorBuffer(filter_out_tensor));

    // Create convolution primitive and add it to net.
    std::vector<primitive> net;
    std::vector<std::unordered_map<int, memory>> net_args;
    if (bias) {
      DCHECK(fuse_biasadd_);
      net.push_back(convolution_forward(conv_prim_desc));
      net_args.push_back({{DNNL_ARG_SRC, src->GetOpMem()},
                          {DNNL_ARG_WEIGHTS, filter->GetOpMem()},
                          {DNNL_ARG_BIAS, bias->GetOpMem()},
                          { DNNL_ARG_DST,
                            output->GetOpMem() }});
    } else {
      DCHECK(!fuse_biasadd_);
      net.push_back(convolution_forward(conv_prim_desc));
      net_args.push_back({{DNNL_ARG_SRC, src->GetOpMem()},
                          {DNNL_ARG_WEIGHTS, filter->GetOpMem()},
                          { DNNL_ARG_DST,
                            output->GetOpMem() }});
    }
    ExecutePrimitive(net, &net_args, cpu_engine_);
  }

  // LOCKS_EXCLUDED annotation ensures that the lock (mu_) cannot
  // be acquired before entering the function, since it is acquired