Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/vulkan/ops/
DUnaryOp.cpp16 const api::ShaderInfo& shader_descriptor) { in unary_op()
62 Tensor& unary_op_(Tensor& self_arg, const api::ShaderInfo& shader_descriptor) { in unary_op_()
Dcumsum.cpp17 api::ShaderInfo& shader_descriptor, in set_cumsum_kernel_params()
146 api::ShaderInfo shader_descriptor; in cumsum() local
DClamp.cpp16 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_()
DSoftmax.cpp17 api::ShaderInfo& shader_descriptor, in set_softmax_kernel_params()
141 api::ShaderInfo shader_descriptor; in softmax_internal() local
DBinaryOp.cpp19 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_()
DPadding.cpp16 const api::ShaderInfo& shader_descriptor) { in pad2d()
DPool.cpp94 const api::ShaderInfo& shader_descriptor) { in pool2d()
/external/pytorch/aten/src/ATen/native/vulkan/api/
DContext.cpp55 const ShaderInfo& shader_descriptor, in get_descriptor_set()
77 const ShaderInfo& shader_descriptor, in register_shader_dispatch()
/external/executorch/backends/vulkan/runtime/api/
DContext.cpp91 const vkapi::ShaderInfo& shader_descriptor, in get_descriptor_set()
121 const vkapi::ShaderInfo& shader_descriptor, in register_shader_dispatch()
DContext.h194 const vkapi::ShaderInfo& shader_descriptor, in get_descriptor_set()
/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DReduce.cpp58 vkapi::ShaderInfo shader_descriptor; in add_reduce_node() local
DSoftmax.cpp59 vkapi::ShaderInfo shader_descriptor; in add_softmax_node() local
/external/pytorch/aten/src/ATen/native/vulkan/impl/
DPacking.cpp302 const api::ShaderInfo& shader_descriptor) { in channel_image_repacking()