Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dfrozen_linear_transpose.cpp61 Tensor weight_t_tensor = at::transpose(weight_tensor, 1, 0) in replace_linear_with_matmul() local
63 Value* weight_t = graph_->insertConstant(std::move(weight_t_tensor)); in replace_linear_with_matmul()