Home
last modified time | relevance | path

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

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