Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp1013 size_t tensor_input_index, in getStrideArg() argument
1016 std::pair<size_t, size_t>(tensor_input_index, stride_index)); in getStrideArg()
1019 "stride_arg" + std::to_string(tensor_input_index) + "_" + in getStrideArg()
1023 tensor_input_index, stride_index)] = var; in getStrideArg()