| /aosp_15_r20/external/pytorch/torch/masked/ |
| H A D | _ops.py | 15 from torch.types import _dtype as DType unknown
20 DType = int variable
620 dtype: Optional[DType] = None,
740 dtype: Optional[DType] = None,
1044 dtype: Optional[DType] = None,
1097 dtype: Optional[DType] = None,
1173 dtype: Optional[DType] = None,
1193 dtype: Optional[DType] = None,
1214 dtype: Optional[DType] = None,
1264 dtype: Optional[DType] = None,
[all …]
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | __init__.py | 20 from torch.types import _dtype as DType unknown
25 DType = int variable
201 def sum(input: Tensor, dim: DimOrDims = None, dtype: Optional[DType] = None) -> Tensor:
|
| /aosp_15_r20/external/pytorch/android/pytorch_android/src/main/java/org/pytorch/ |
| H A D | DType.java | 4 public enum DType { enum
24 DType(int jniCode) { in DType() method in DType
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | dynamic_ragged_shape.py | 199 dtype: Optional[dtypes.DType] = None,
1119 dtype: dtypes.DType):
1240 dtype: dtypes.DType) -> "DynamicRaggedShape.Spec":
1311 dtype: dtypes.DType = dtypes.int64) -> "DynamicRaggedShape.Spec":
1434 def with_dtype(self, dtype: dtypes.DType) -> "DynamicRaggedShape.Spec":
1882 dtype: Optional[dtypes.DType]) -> Optional[dtypes.DType]:
|
| /aosp_15_r20/external/executorch/extension/android/src/main/java/org/pytorch/executorch/ |
| H A D | DType.java | 19 public enum DType { enum
73 DType(int jniCode) { in DType() method in DType
|
| /aosp_15_r20/external/pytorch/torch/nested/ |
| H A D | __init__.py | 9 from torch.types import _device as Device, _dtype as DType unknown
25 dtype: Optional[DType] = None,
|
| /aosp_15_r20/external/executorch/extension/llm/export/ |
| H A D | builder.py | 44 class DType(Enum): class
121 def to_dtype(self, dtype_override: Optional[DType]) -> "LLMEdgeManager":
|
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | export_llama_lib.py | 867 dtype_override: Optional[DType] = None,
979 modelname: str, dtype_override: Optional[DType], args
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 30 from torch.types import _dtype as DType unknown
33 DType = int variable
2074 dtype: Optional[DType] = None,
2107 dtype: Optional[DType] = None,
2224 dtype: Optional[DType] = None,
5835 other_type: Optional[DType],
5837 target_type: DType,
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | convert.py | 63 def convert_tensor_tf_type_to_tflite_type(tf_type: dtypes.DType,
105 def convert_inference_tf_type_to_tflite_type(tf_type: dtypes.DType,
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/IR/ |
| H A D | DebugInfo.cpp | 1227 const char *LLVMDITypeGetName(LLVMMetadataRef DType, size_t *Length) { in LLVMDITypeGetName()
1233 uint64_t LLVMDITypeGetSizeInBits(LLVMMetadataRef DType) { in LLVMDITypeGetSizeInBits()
1237 uint64_t LLVMDITypeGetOffsetInBits(LLVMMetadataRef DType) { in LLVMDITypeGetOffsetInBits()
1241 uint32_t LLVMDITypeGetAlignInBits(LLVMMetadataRef DType) { in LLVMDITypeGetAlignInBits()
1245 unsigned LLVMDITypeGetLine(LLVMMetadataRef DType) { in LLVMDITypeGetLine()
1249 LLVMDIFlags LLVMDITypeGetFlags(LLVMMetadataRef DType) { in LLVMDITypeGetFlags()
|
| /aosp_15_r20/external/federated-compute/fcp/demo/ |
| H A D | test_utils.py | 36 dt: tf.DType) -> Any:
|
| H A D | checkpoint_tensor_reference.py | 26 def __init__(self, tensor_name: str, dtype: tf.DType, shape: Any,
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/IR/ |
| H A D | DebugInfo.cpp | 1447 const char *LLVMDITypeGetName(LLVMMetadataRef DType, size_t *Length) { in LLVMDITypeGetName()
1453 uint64_t LLVMDITypeGetSizeInBits(LLVMMetadataRef DType) { in LLVMDITypeGetSizeInBits()
1457 uint64_t LLVMDITypeGetOffsetInBits(LLVMMetadataRef DType) { in LLVMDITypeGetOffsetInBits()
1461 uint32_t LLVMDITypeGetAlignInBits(LLVMMetadataRef DType) { in LLVMDITypeGetAlignInBits()
1465 unsigned LLVMDITypeGetLine(LLVMMetadataRef DType) { in LLVMDITypeGetLine()
1469 LLVMDIFlags LLVMDITypeGetFlags(LLVMMetadataRef DType) { in LLVMDITypeGetFlags()
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/ |
| H A D | signature.go | 89 DType DataType member
|
| /aosp_15_r20/external/federated-compute/fcp/artifact_building/ |
| H A D | tensor_utils.py | 117 tensor_type: tf.DType,
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/ |
| H A D | tensor_view.h | 207 const std::size_t data_size, const DType &) in TensorView()
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/CodeGen/ |
| H A D | JMCInstrumenter.cpp | 110 auto *DType = in attachDebugInfo() local
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
| H A D | test_dtype.py | 217 def test_pickle_types(self, DType): argument
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/clang/Basic/ |
| D | OpenACCKinds.h | 252 DType, enumerator
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/clang/Basic/ |
| D | OpenACCKinds.h | 252 DType, enumerator
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/clang/Basic/ |
| D | OpenACCKinds.h | 277 DType, enumerator
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | dtypes.py | 42 class DType( class
|
| /aosp_15_r20/external/executorch/examples/models/llama/source_transformation/ |
| H A D | quantize.py | 41 activation_dtype: Optional[DType],
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _dtypes.py | 274 class DType: class
|