Home
last modified time | relevance | path

Searched refs:flatInputTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DView.mm267 MPSGraphTensor* flatInputTensor = inputTensor;
271 for (NSUInteger srcDim = 1; srcDim < [[flatInputTensor shape] count]; srcDim++) {
272 if ([[flatInputTensor shape][srcDim] intValue] == 1)
277 flatInputTensor = [graph squeezeTensor:flatInputTensor axes:squeezeAxes name:nil];
281 int srcRank = (int)[[flatInputTensor shape] count];
282 …NSDictionary* srcStrideToDimLengthOffset = getStrideToDimLengthOffsetDict(flatInputTensor, srcRank…
321 MPSGraphTensor* slicedUnusedTensor = flatInputTensor;
327 for (NSInteger srcDim = [[flatInputTensor shape] count] - 1; srcDim >= 0; srcDim--) {
329 stride *= [[flatInputTensor shape][srcDim] integerValue];
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupMPI.cpp642 at::Tensor flatInputTensor; in scatter() local
646 flatInputTensor = newLikeFlat(inputDataVec); in scatter()
647 sendbuf = flatInputTensor.data_ptr(); in scatter()
651 flatInputTensor[static_cast<int64_t>(i)].copy_(inputDataVec.at(i)); in scatter()
H A DProcessGroupGloo.cpp1829 at::Tensor flatInputTensor = flattenDenseTensors(inputs); in allgather() local
1830 GENERATE_ALL_TYPES(scalarType, setInput, opts, flatInputTensor); in allgather()
2103 at::Tensor flatInputTensor = flattenDenseTensors(input_list); in allgather_coalesced() local
2104 GENERATE_ALL_TYPES(scalarType, setInput, opts, flatInputTensor); in allgather_coalesced()