Home
last modified time | relevance | path

Searched defs:TensorMetadata (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/
H A Dkernel_meta_info.h31 struct TensorMetadata { struct
52 TensorMetadata() in TensorMetadata() argument
H A Dkernel_meta_info.cpp8 TensorMetadata::TensorMetadata(const at::Tensor& src_tensor) in TensorMetadata() function in torch::inductor::TensorMetadata
17 TensorMetadata::TensorMetadata( in TensorMetadata() function in torch::inductor::TensorMetadata
/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dcollection.h70 struct TORCH_API TensorMetadata : public RawTensorMetadataBase { struct
76 TensorImplAddress impl() const { in impl()
102 TensorMetadata, argument
H A Dcollection.cpp55 TensorMetadata::TensorMetadata( in TensorMetadata() function in torch::profiler::impl::TensorMetadata
/aosp_15_r20/external/executorch/exir/passes/
H A D__init__.py266 Optional[TensorMetadata],
267 List[Optional[TensorMetadata]],
268 Tuple[Optional[TensorMetadata]],
/aosp_15_r20/external/executorch/extension/flat_tensor/
H A Dflat_tensor_schema.py17 class TensorMetadata: class
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dhost_compute_metadata.proto14 message TensorMetadata { message
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dsnapshot.proto38 message TensorMetadata { message
/aosp_15_r20/external/pytorch/torch/fx/passes/
H A Dshape_prop.py17 class TensorMetadata(NamedTuple): class
H A Dgraph_drawer.py312 def _stringify_tensor_meta(self, tm: TensorMetadata) -> str:
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dmobile_bytecode_generated.h144 TensorMetadata = 5, enumerator
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.h62 class TensorMetadata; variable
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_tensor.py912 class TensorMetadata: class