| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Activation.mm | 204 …MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0f shape:@[ @1 ] dataType:getMPSScalar…
206 secondaryTensor:zeroTensor
338 …MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 shape:@[ @1 ] dataType:inputTensor.d…
339 …r* minTensor = [mpsGraph minimumWithPrimaryTensor:inputTensor secondaryTensor:zeroTensor name:nil];
398 …MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 shape:@[ @1 ] dataType:inputTensor.d…
402 secondaryTensor:zeroTensor
406 falsePredicateTensor:zeroTensor
599 MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 dataType:inputTensor.dataType];
606 // result = (self > threshold) ? grad : zeroTensor
609 falsePredicateTensor:zeroTensor
[all …]
|
| H A D | UnaryOps.mm | 181 MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 dataType:inputTensor.dataType];
182 output = [mpsGraph lessThanWithPrimaryTensor:inputTensor secondaryTensor:zeroTensor name:nil];
301 auto zeroTensor = [mpsGraph constantWithScalar:0.0 dataType:inputTensor.dataType];
302 …icateTensor = [mpsGraph lessThanWithPrimaryTensor:inputTensor secondaryTensor:zeroTensor name:nil];
379 …MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 shape:@[ @1 ] dataType:inputTensor.d…
400 truePredicateTensor:zeroTensor
|
| H A D | Unique.mm | 132 …MPSGraphTensor* zeroTensor = [graph constantWithScalar:0.0f shape:@[ @1 ] dataType:MPSDataTypeInt3…
133 …MPSGraphTensor* maskedIndicesWithHead = [graph concatTensors:@[ zeroTensor, maskedIndices ] dimens…
134 …MPSGraphTensor* scannedIndicesWithHead = [graph concatTensors:@[ zeroTensor, scannedIndices ] dime…
|
| H A D | BinaryOps.mm | 516 …MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 shape:@[ @1 ] dataType:primaryCastTe…
523 secondaryTensor:zeroTensor
526 truePredicateTensor:zeroTensor
|
| H A D | Distributions.mm | 484 auto zeroTensor = [mpsGraph constantWithScalar:0.0f dataType:MPSDataTypeInt32];
488 numLowerTensor:zeroTensor
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ZeroTensorFallback.cpp | 26 "ZeroTensor fallback doesn't work for operators with a mix " in zeroTensorFallback()
37 // We assume that view operators automatically handle the ZeroTensor bit in zeroTensorFallback()
40 …d(dispatch_keys & c10::DispatchKeySet(DispatchKeySet::FULL_AFTER, DispatchKey::ZeroTensor), stack); in zeroTensorFallback()
72 //_like should not propagate zerotensor dispatch key in zeroTensorFallback()
82 …d(dispatch_keys & c10::DispatchKeySet(DispatchKeySet::FULL_AFTER, DispatchKey::ZeroTensor), stack); in zeroTensorFallback()
86 TORCH_LIBRARY_IMPL(_, ZeroTensor, m) { in TORCH_LIBRARY_IMPL() argument
90 TORCH_LIBRARY_IMPL(aten, ZeroTensor, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | legalize_to_linalg.cc | 1445 Value zeroTensor = fillTensorWithZeros(rewriter, loc, initTensor); in matchAndRewrite() local
1448 ValueRange{zeroTensor}, pruneAttributeList(op)); in matchAndRewrite()
1496 Value zeroTensor = fillTensorWithZeros(rewriter, loc, initTensor); in matchAndRewrite() local
1500 /*outputBuffers=*/ValueRange{zeroTensor}, pruneAttributeList(op)); in matchAndRewrite()
1907 Value zeroTensor = fillTensorWithZeros(rewriter, loc, initTensor); in matchAndRewrite() local
1922 loc, resultType, ValueRange{input, filter}, ValueRange{zeroTensor}, in matchAndRewrite()
1928 loc, resultType, ValueRange{input, filter}, ValueRange{zeroTensor}, in matchAndRewrite()
1934 loc, resultType, ValueRange{input, filter}, ValueRange{zeroTensor}, in matchAndRewrite()
1940 loc, resultType, ValueRange{input, filter}, ValueRange{zeroTensor}, in matchAndRewrite()
2227 Value zeroTensor = fillTensorWithZeros(rewriter, loc, initTensor); in matchAndRewrite() local
[all …]
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.cpp | 122 case DispatchKey::ZeroTensor: in toString()
123 return "ZeroTensor"; in toString()
286 {"ZeroTensor", c10::DispatchKey::ZeroTensor}, in parseDispatchKey()
|
| H A D | TensorImpl.h | 1412 * Whether or not the tensor is a zerotensor
1415 constexpr auto zerotensor_ks = DispatchKeySet(DispatchKey::ZeroTensor); in _is_zerotensor()
1428 key_set_ = key_set_.remove(DispatchKey::ZeroTensor); in _set_zero()
|
| H A D | DispatchKey.h | 273 ZeroTensor, // registered at build/aten/src/ATen/RegisterZeroTensor.cpp enumerator
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | functional_tensor.py | 71 # except that they don't include ZeroTensor so I'm manually adding it in.
73 torch._C.DispatchKey.ZeroTensor
123 …# Example: If our inner tensor is a ZeroTensor, we would want to defer running the ZeroTensor fall…
124 # until after we redispatch to our inner ZeroTensor.
|
| /aosp_15_r20/external/executorch/backends/apple/mps/runtime/operations/ |
| H A D | IndexingOps.mm | 91 MPSGraphTensor* zeroTensor = [_mpsGraph constantWithScalar:0
97 falsePredicateTensor:zeroTensor
|
| H A D | BinaryOps.mm | 182 MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 dataType:inputTensor.dataType];
184 secondaryTensor:zeroTensor
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | pickler.h | 356 // We don't support serializing `ZeroTensor` as it is not public in getTensorMetadata()
360 "ZeroTensor is not serializable,", in getTensorMetadata()
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_gen.py | 52 ZeroTensor: add_zerotensor
76 ZeroTensor: mul_zerotensor
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | serialize.cpp | 299 // We don't support serializing `ZeroTensor` as it is not public facing yet. in TEST()
300 // If in future, `ZeroTensor` serialization is supported, this test should in TEST()
303 ASSERT_THROWS_WITH(save_and_load(t), "ZeroTensor is not serializable,"); in TEST()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/ |
| H A D | TensorFactory.cpp | 161 …_ks = at::DispatchKeySet(c10::DispatchKey::MPS) | at::DispatchKeySet(c10::DispatchKey::ZeroTensor); in _efficientzerotensor_mps()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorFactories.h | 119 // The ZeroTensor allocator ignores whatever allocation is requested and always
|
| H A D | TensorFactories.cpp | 416 // never propagate Conjugate, Negative, and ZeroTensor dispatch key in AT_FORALL_SCALAR_TYPES_AND3()
1329 …_ks = at::DispatchKeySet(c10::DispatchKey::CPU) | at::DispatchKeySet(c10::DispatchKey::ZeroTensor); in _efficientzerotensor()
1342 …ks = at::DispatchKeySet(c10::DispatchKey::Meta) | at::DispatchKeySet(c10::DispatchKey::ZeroTensor); in _efficientzerotensor_meta_symint()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_python_dispatch.py | 232 my_lib2.impl("aten::mul.Tensor", my_mul, "ZeroTensor")
242 my_lib2.impl(torch.ops.aten.mul.Tensor, my_mul, "ZeroTensor")
473 @torch.library.impl(my_lib2, op.default, "ZeroTensor")
|
| H A D | test_fx.py | 1466 class ZeroTensor(metaclass=torch.fx.ProxyableClassMeta): class
1477 return ZeroTensor(other.tensor)
1482 return ZeroTensor(x + y)
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorFactories.cu | 74 …ks = at::DispatchKeySet(c10::DispatchKey::CUDA) | at::DispatchKeySet(c10::DispatchKey::ZeroTensor); in _efficientzerotensor_cuda()
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 103 ZeroTensor = auto() variable in DispatchKey
299 DispatchKey.ZeroTensor,
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 713 DEF_ONE(ZeroTensor) in initDispatchBindings()
|
| /aosp_15_r20/external/pytorch/docs/source/notes/ |
| H A D | extending.rst | 898 vmap -> Autocast -> Autograd -> ZeroTensor -> Neg/Conj -> Functionalize -> Python -> Backends
|