Home
last modified time | relevance | path

Searched refs:extract_tensor_meta (Results 1 – 5 of 5) sorted by relevance

/external/executorch/backends/arm/_passes/
Dconvert_expand_copy_to_repeat.py11 from executorch.backends.arm.tosa_mapping import extract_tensor_meta
28 _, shape, _ = extract_tensor_meta(meta.data)
Dconvert_split_to_slice.py11 from executorch.backends.arm.tosa_mapping import extract_tensor_meta
37 _, shape, _ = extract_tensor_meta(input_node.meta)
/external/executorch/backends/arm/
Dtosa_mapping.py54 def extract_tensor_meta(meta): function
77 self.dtype, self.shape, self.dim_order = extract_tensor_meta(argument.meta)
/external/pytorch/torch/fx/passes/
Dshape_prop.py166 def extract_tensor_meta(obj): function
174 meta = map_aggregate(result, extract_tensor_meta)
/external/executorch/backends/arm/test/tester/
Darm_tester.py38 from executorch.backends.arm.tosa_mapping import extract_tensor_meta
557 dtype, _, _ = extract_tensor_meta(node.meta)