Searched defs:reference_out (Results 1 – 3 of 3) sorted by relevance
70 auto reference_out = model.GetDequantizedOutput<integer_dtype>(); in TestMulOutputImpl() local89 auto reference_out = model.GetDequantizedOutput<integer_dtype>(); in TestLargeInputRangeImpl() local
247 at::Tensor reference_out = torch::executor::native::sdpa_with_kv_cache_aten( in test_reference_sdpa() local397 at::Tensor reference_out = sdpa_reference_impl( in test_vulkan_sdpa() local
794 void set_sizes_strides_offset(const Tensor& out, const Tensor& reference_out) { in set_sizes_strides_offset()