Home
last modified time | relevance | path

Searched defs:TensorSpec (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/executorch/exir/
H A Dmemory_planning.py55 cls, lhs_spec: TensorSpec, rhs_spec: TensorSpec, accept_both_none: bool = True
86 def lifetime_overlap(cls, lhs_spec: TensorSpec, rhs_spec: TensorSpec) -> bool:
100 def storage_overlap(cls, lhs_spec: TensorSpec, rhs_spec: TensorSpec) -> bool:
271 def update_tensor_lifetime(spec: TensorSpec, node_idx: int) -> None:
487 def _size_abs_dif(sobj: SharedObject, spec: TensorSpec) -> int:
496 shared_objects: List[SharedObject], spec: TensorSpec
683 graph_module: fx.GraphModule, allspecs: Set[TensorSpec]
H A Dtensor.py122 class TensorSpec: class
319 spec: TensorSpec,
353 def check_spec(tensor: torch.Tensor, spec: TensorSpec) -> None:
H A Dmemory.py36 def free(spec: TensorSpec) -> None:
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.h52 struct TensorSpec { struct
56 explicit TensorSpec(tensorflow::DataType dtype) : dtype(dtype) {} in TensorSpec() argument
57 TensorSpec(tensorflow::DataType dtype, tensorflow::PartialTensorShape shape) in TensorSpec() function
319 using TensorSpec = ::tensorflow::tfrt_stub::TensorSpec; variable
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/
H A Ddepthfirst_driver.hpp47 struct TensorSpec struct
49 T base;
50 size_t ld_row, ld_col;
52 TensorSpec(T ptr, size_t ld_row, size_t ld_col) in TensorSpec() argument
/aosp_15_r20/external/executorch/test/end2end/
H A Dregister_scratch_meta_fns.py30 input: TensorSpec,
31 weight: TensorSpec,
32 bias: Optional[TensorSpec],
34 out: TensorSpec,
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
H A Ddepthfirst_driver.hpp55 struct TensorSpec struct
57 T base;
58 size_t ld_row, ld_col;
60 TensorSpec(T ptr, size_t ld_row, size_t ld_col) in TensorSpec() function
/aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libtf/impl/
H A Dtensor_spec.h30 struct TensorSpec { struct
34 bool operator==(const TensorSpec& o) const { argument
40 friend H AbslHashValue(H h, const TensorSpec& t) { in AbslHashValue() argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/coreml/objc/
H A DPTMCoreMLTensorSpec.h11 struct TensorSpec { struct
12 std::string name = "";
13 c10::ScalarType dtype = c10::ScalarType::Float;
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/
H A Dtensor_spec.cc25 TensorSpec::TensorSpec() in TensorSpec() function in tensorflow::TensorSpec
28 TensorSpec::TensorSpec(PartialTensorShape shape, DataType dtype) in TensorSpec() function in tensorflow::TensorSpec
31 TensorSpec::TensorSpec(const TensorSpecProto& proto) in TensorSpec() function in tensorflow::TensorSpec
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/models/
H A Dlog_reader.py27 class TensorSpec: class
50 def __init__(self, spec: TensorSpec, buffer: bytes):
69 def read_tensor(fs: typing.BinaryIO, ts: TensorSpec) -> TensorValue:
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/
H A Dinput_util.py107 element_spec: tensor_spec.TensorSpec, layouts: Any,
211 flattened_elem_spec: Sequence[tensor_spec.TensorSpec],
292 elem_spec: tensor_spec.TensorSpec) -> ops.Tensor:
/aosp_15_r20/external/executorch/exir/emit/
H A D_emitter.py141 def spec2id(self, spec: TensorSpec) -> int:
331 def _tensor_spec_to_evalue(self, spec: TensorSpec) -> EValue:
994 spec: TensorSpec,
1206 def _emit_free(self, spec: TensorSpec) -> _AbstractValue:
/aosp_15_r20/external/executorch/exir/operator/
H A Dmanip.py63 def wrapper(*args: TensorSpec, **kwargs: TensorSpec) -> Dict[str, TensorSpec]:
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dmemory_planning.py67 def overlap(spec: TensorSpec) -> Optional[TensorSpec]:
75 def memory_available(spec: TensorSpec) -> bool:
/aosp_15_r20/external/executorch/exir/passes/
H A Ddynamic_shape_prop_pass.py85 def copy_dsinfo_btw_specs(self, src_spec: TensorSpec, dst_spec: TensorSpec):
H A Dreplace_view_copy_with_view_pass.py57 def __init__(self, base: TensorSpec, shape: List[int]) -> None:
/aosp_15_r20/external/federated-compute/fcp/aggregation/core/
H A Dtensor_spec.h32 TensorSpec(std::string name, DataType dtype, TensorShape shape) in TensorSpec() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/llvm/Analysis/
DTensorSpec.h90 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/Analysis/
H A DTensorSpec.h86 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/llvm/Analysis/
DTensorSpec.h90 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/llvm/Analysis/
DTensorSpec.h90 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/llvm/Analysis/
DTensorSpec.h90 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
/aosp_15_r20/external/pytorch/torch/backends/_coreml/
H A Dpreprocess.py48 def TensorSpec(shape, dtype=ScalarType.Float): function
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/
H A DTensorSpec.cpp60 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type, in TensorSpec() function in llvm::TensorSpec

12