Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/vulkan/ops/
DLstm.cpp92 const Tensor& cx_vk = hx[1]; in lstm_input() local
325 const Tensor& cx_vk, // initial cell state (vulkan) in run_lstm_context()