Home
last modified time | relevance | path

Searched refs:cat_compute_output_memory_format (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp229 inline c10::MemoryFormat cat_compute_output_memory_format(const MaterializedITensorListRef& inputs)… in cat_compute_output_memory_format() function
272 auto memory_format = cat_compute_output_memory_format(materialized); in TORCH_PRECOMPUTE_META_FUNC()
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py2718 def cat_compute_output_memory_format(inputs): function
2805 memory_format = cat_compute_output_memory_format(tensors)