Searched refs:sliceParams (Results 1 – 1 of 1) sorted by relevance
179 explicit MklSlicePrimitive(const MklSliceParams& sliceParams) in MklSlicePrimitive() argument181 Setup(sliceParams); in MklSlicePrimitive()186 void Execute(const MklSliceParams& sliceParams, in Execute() argument189 context_.src_mem->set_data_handle(sliceParams.from->get_data_handle(), in Execute()191 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle(), in Execute()194 context_.src_mem->set_data_handle(sliceParams.from->get_data_handle()); in Execute()195 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle()); in Execute()226 void Setup(const MklSliceParams& sliceParams) { in Setup() argument230 new memory(sliceParams.from->get_desc(), cpu_engine_, DummyData)); in Setup()232 new memory(sliceParams.to->get_desc(), cpu_engine_, DummyData)); in Setup()[all …]