Method Execute

tensorflow/core/kernels/mkl_slice_op.cc:190–214

  ~MklSlicePrimitive() {}

  void Execute(const MklSliceParams& sliceParams,
               std::shared_ptr<stream> slice_stream) {
#ifdef DNNL_AARCH64_USE_ACL
    mutex_lock lock(primitive_execution_mu_);
#endif
#ifdef ENABLE_DNNL_THREADPOOL
    context_.src_mem->set_data_handle(sliceParams.from->get_data_handle(),
                                      *slice_stream);
    context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle(),
                                      *slice_stream);
#else
    context_.src_mem->set_data_handle(sliceParams.from->get_data_handle());
    context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle());
#endif  // ENABLE_DNNL_THREADPOOL
    execute_primitives(context_.slice_primitives, slice_stream,
                       context_.slice_primitives_args);

    // We should set it back to DummyData so as to make the primitive
    // in cache pool stateless. Otherwise, if the result for previous
    // iteration is kept, problems of current iteration won't be
    // thrown immediately, and wrong data would be reused.
    context_.src_mem->set_data_handle(DummyData);
    context_.dst_mem->set_data_handle(DummyData);
    return;
  }

  std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; }