Home
last modified time | relevance | path

Searched refs:mpsGraphTensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DOperationUtils.h103 …Placeholder(MPSGraphTensor* mpsGraphTensor) : _placeholder(mpsGraphTensor), _value(nullptr), _tens… in Placeholder() argument
104 Placeholder(MPSGraphTensor* mpsGraphTensor, MPSNDArray* mpsNDArray);
105 Placeholder(MPSGraphTensor* mpsGraphTensor, const Tensor& self, MPSShape *mpsShape = nullptr,
H A DOperationUtils.mm488 Placeholder::Placeholder(MPSGraphTensor* mpsGraphTensor, MPSNDArray* mpsNDArray) {
489 _placeholder = mpsGraphTensor;
493 Placeholder::Placeholder(MPSGraphTensor* mpsGraphTensor,
582 _placeholder = mpsGraphTensor;