Searched defs:input_vk (Results 1 – 4 of 4) sorted by relevance
43 const Tensor& input_vk, // input sequence (vulkan) in gru_input()281 const Tensor& input_vk, // input sequence (vulkan) in run_gru_context()
51 const Tensor& input_vk, // input sequence (vulkan) in lstm_input()323 const Tensor& input_vk, // input sequence (vulkan) in run_lstm_context()
1393 const auto input_vk = input_cpu.vulkan(); in TEST_F() local1435 const auto input_vk = input_cpu.vulkan(); in test_conv1d() local2131 const auto input_vk = at::div(input_numerator.vulkan(), input_denominator.vulkan()); in TEST_F() local
2509 at::Tensor input_vk; // input cpu tensor -> to vulkan in test_quantized_conv2d() local