Home
last modified time | relevance | path

Searched refs:IntToTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dconstant_fold.h19 at::Tensor IntToTensor(int64_t value);
H A Dconstant_fold.cpp251 at::Tensor IntToTensor(int64_t value) { in IntToTensor() function
535 return std::optional<at::Tensor>(IntToTensor(total_size)); in runTorchBackendForOnnx()
H A Dshape_type_inference.cpp1492 auto f_final = onnx_constant_fold::IntToTensor(total_size); in ComputeConstant()