Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_concat_linear.cpp48 std::vector<Value*>& ordered_tensor_inputs) { in collectConstantLinearLayers() argument
78 ordered_tensor_inputs.push_back(linear_input); in collectConstantLinearLayers()
227 std::vector<Value*> ordered_tensor_inputs; in handleBlockAndSubblocks() local
229 block, grouped_linear_layers, ordered_tensor_inputs); in handleBlockAndSubblocks()
233 for (auto tensor_it = ordered_tensor_inputs.rbegin(); in handleBlockAndSubblocks()
234 tensor_it != ordered_tensor_inputs.rend(); in handleBlockAndSubblocks()