File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
aten/src/ATen/native/mps/operations Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 322322 Placeholder outputPlaceholder3 = Placeholder (cachedGraph->outputTensors_ [3 ], zState);
323323 Placeholder outputPlaceholder4 = Placeholder (cachedGraph->outputTensors_ [4 ], cellStateFwd);
324324
325- NSMutableDictionary <MPSGraphTensor*, MPSGraphTensorData*>* results = [@{
325+ NSMutableDictionary <MPSGraphTensor*, MPSGraphTensorData*>* results = [[ @{
326326 outputPlaceholder0.getMPSGraphTensor () : outputPlaceholder0.getMPSGraphTensorData (),
327327 outputPlaceholder1.getMPSGraphTensor () : outputPlaceholder1.getMPSGraphTensorData (),
328328 outputPlaceholder2.getMPSGraphTensor () : outputPlaceholder2.getMPSGraphTensorData (),
329329 outputPlaceholder3.getMPSGraphTensor () : outputPlaceholder3.getMPSGraphTensorData (),
330330 outputPlaceholder4.getMPSGraphTensor () : outputPlaceholder4.getMPSGraphTensorData (),
331- } mutableCopy];
331+ } mutableCopy] autorelease ] ;
332332
333333 if (num_layers > 1 ) {
334334 Placeholder outputPlaceholder5 = Placeholder (cachedGraph->outputTensors_ [5 ], layerOutputs);
You can’t perform that action at this time.
0 commit comments