Searched refs:slice_stream (Results 1 – 1 of 1) sorted by relevance
187 std::shared_ptr<stream> slice_stream) { in Execute() argument190 *slice_stream); in Execute()192 *slice_stream); in Execute()198 execute_primitives(context_.slice_primitives, slice_stream, in Execute()218 std::shared_ptr<mkldnn::stream> slice_stream; member455 std::shared_ptr<stream> slice_stream; in ComputeMklSlice() local456 slice_stream.reset(CreateStream(context, reorder_prim->GetEngine())); in ComputeMklSlice()457 reorder_prim->Execute(sliceParams, slice_stream); in ComputeMklSlice()