Searched defs:shader_descriptor (Results 1 – 13 of 13) sorted by relevance
16 const api::ShaderInfo& shader_descriptor) { in unary_op()62 Tensor& unary_op_(Tensor& self_arg, const api::ShaderInfo& shader_descriptor) { in unary_op_()
17 api::ShaderInfo& shader_descriptor, in set_cumsum_kernel_params()146 api::ShaderInfo shader_descriptor; in cumsum() local
16 const api::ShaderInfo& shader_descriptor) { in _clamp()108 const api::ShaderInfo& shader_descriptor) { in _clamp_()196 const api::ShaderInfo& shader_descriptor) { in activation()244 const api::ShaderInfo& shader_descriptor) { in activation_()335 const api::ShaderInfo& shader_descriptor) { in activation_scalar()426 const api::ShaderInfo& shader_descriptor) { in activation_scalar_()
17 api::ShaderInfo& shader_descriptor, in set_softmax_kernel_params()141 api::ShaderInfo shader_descriptor; in softmax_internal() local
19 const api::ShaderInfo& shader_descriptor) { in binary_op_scalar()106 const api::ShaderInfo& shader_descriptor) { in binary_op_scalar_()156 const api::ShaderInfo& shader_descriptor) { in binary_op_tensor()230 const api::ShaderInfo& shader_descriptor) { in quantized_binary_op_tensor()317 const api::ShaderInfo& shader_descriptor) { in binary_op_tensor_()
16 const api::ShaderInfo& shader_descriptor) { in pad2d()
94 const api::ShaderInfo& shader_descriptor) { in pool2d()
55 const ShaderInfo& shader_descriptor, in get_descriptor_set()77 const ShaderInfo& shader_descriptor, in register_shader_dispatch()
91 const vkapi::ShaderInfo& shader_descriptor, in get_descriptor_set()121 const vkapi::ShaderInfo& shader_descriptor, in register_shader_dispatch()
194 const vkapi::ShaderInfo& shader_descriptor, in get_descriptor_set()
58 vkapi::ShaderInfo shader_descriptor; in add_reduce_node() local
59 vkapi::ShaderInfo shader_descriptor; in add_softmax_node() local
302 const api::ShaderInfo& shader_descriptor) { in channel_image_repacking()