Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSamplerUtils.h26 auto input_opt = input.options(); in check_grid_sampler_common() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseSemiStructuredOps.cu525 const std::optional<Tensor>& input_opt, const Scalar& alpha, in sparse_semi_structured_mad_op()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dpasses.cpp1135 auto input_opt = toIValue(node->input(input_idx)); in inputIsConstantList() local
1143 auto input_opt = toIValue(node->input(input_idx)); in inputIsConstantInt() local
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Drnn.cpp577 auto input_opt = in TEST_F() local