Home
last modified time | relevance | path

Searched defs:TensorMeta (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dautotune_process.py417 class TensorMeta: class
482 input_tensor_meta: Union[TensorMeta, List[TensorMeta]],
483 output_tensor_meta: Union[TensorMeta, List[TensorMeta]],
610 input_tensor_meta: Union[TensorMeta, List[TensorMeta]],
611 output_tensor_meta: Union[TensorMeta, List[TensorMeta]],
687 input_tensor_meta: Union[TensorMeta, List[TensorMeta]],
688 output_tensor_meta: Union[TensorMeta, List[TensorMeta]],
814 input_tensor_meta: Union[TensorMeta, List[TensorMeta]],
815 output_tensor_meta: Union[TensorMeta, List[TensorMeta]],
/aosp_15_r20/external/executorch/runtime/kernel/
H A Doperator_registry.h52 struct TensorMeta { struct
56 TensorMeta() = default; argument
57 TensorMeta( in TensorMeta() argument
70 bool equals(const TensorMeta& other) const { in equals() argument
101 * `TensorMeta` are matched. Fallback means this kernel will be used for argument
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Drocm_benchmark_request.py24 input_tensor_meta: Union[TensorMeta, List[TensorMeta]],
25 output_tensor_meta: Union[TensorMeta, List[TensorMeta]],
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_dtensor_spec.py14 class TensorMeta(NamedTuple): class
214 tensor_meta: Optional[TensorMeta] = None,
266 self, tensor_meta: Optional[TensorMeta]
H A D_sharding_prop.py157 output_tensor_meta: Union[None, TensorMeta, Sequence[Optional[TensorMeta]]], argument
472 out_tensor_meta: TensorMeta,
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dschema.py87 class TensorMeta: class
H A Dserialize.py1574 tensor_meta: TensorMeta,
/aosp_15_r20/external/executorch/exir/serde/
H A Dschema.py95 class TensorMeta: class
H A Dexport_serialize.py1547 tensor_meta: TensorMeta,
/aosp_15_r20/external/pytorch/torch/_prims/
H A D__init__.py223 def TensorMeta( function