Home
last modified time | relevance | path

Searched defs:opt_size (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DMetaTensor.cpp23 auto opt_size = asIntArrayRefSlowOpt(size); in empty_meta_symint() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DNestedTensorImpl.cpp239 std::optional<int64_t> NestedTensorImpl::opt_size(int64_t d) const { in opt_size() function in at::native::NestedTensorImpl
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMatmul.cpp320 std::optional<int64_t> opt_size = function_result_ptr->opt_size(i); in matmul_out_nested() local
H A DNestedTensorMath.cpp965 std::optional<int64_t> opt_size = other_ptr->opt_size(i); in reshape_as_nested() local