Searched refs:sliceParams (Results 1 – 1 of 1) sorted by relevance
181 explicit MklSlicePrimitive(const MklSliceParams& sliceParams) { in MklSlicePrimitive() argument183 Setup(sliceParams); in MklSlicePrimitive()188 void Execute(const MklSliceParams& sliceParams) { in Execute() argument189 context_.src_mem->set_data_handle(sliceParams.from->get_data_handle()); in Execute()190 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle()); in Execute()219 void Setup(const MklSliceParams& sliceParams) { in Setup() argument223 new memory({sliceParams.from->get_primitive_desc().desc(), cpu_engine_}, in Setup()226 {sliceParams.to->get_primitive_desc().desc(), cpu_engine_}, DummyData)); in Setup()231 src_pd, sliceParams.size_dims, sliceParams.begin_dims)); in Setup()244 static MklSlicePrimitive<T>* Get(const MklSliceParams& sliceParams) { in Get() argument[all …]