| /external/executorch/backends/vulkan/runtime/graph/ |
| D | ComputeGraph.cpp | 25 classname::classname(ComputeGraph* const graph, const ValueRef idx) \
55 ComputeGraph* const graph_ptr, in VALUE_PTR_CLASS_IMPL()
70 ComputeGraph* const graph_ptr, in TmpTensor()
79 ComputeGraph* const graph_ptr, in TmpTensor()
88 ComputeGraph* const graph_ptr, in TmpTensor()
120 ComputeGraph::ComputeGraph(GraphConfig config) in ComputeGraph() function in vkcompute::ComputeGraph
150 ComputeGraph::~ComputeGraph() { in ~ComputeGraph()
159 utils::StorageType ComputeGraph::suggested_storage_type() { in suggested_storage_type()
166 utils::GPUMemoryLayout ComputeGraph::suggested_memory_layout( in suggested_memory_layout()
182 void ComputeGraph::check_no_active_value_ptrs() { in check_no_active_value_ptrs()
[all …]
|
| D | ComputeGraph.h | 46 class ComputeGraph;
50 ComputeGraph* const graph_; \
54 explicit classname(ComputeGraph* const graph, const ValueRef idx); \
91 ComputeGraph* graph_p;
100 ComputeGraph* const graph_ptr,
107 ComputeGraph* const graph_ptr,
113 ComputeGraph* const graph_ptr,
119 ComputeGraph* const graph_ptr,
156 class ComputeGraph final {
158 explicit ComputeGraph(GraphConfig config);
[all …]
|
| /external/executorch/backends/vulkan/runtime/graph/containers/ |
| D | SharedObject.h | 23 class ComputeGraph; variable
26 friend class ComputeGraph; member
35 void add_user(ComputeGraph* const graph, const ValueRef idx);
36 void allocate(ComputeGraph* const graph);
37 void bind_users(ComputeGraph* const graph);
|
| D | SharedObject.cpp | 19 void SharedObject::add_user(ComputeGraph* const graph, const ValueRef idx) { in add_user()
33 void SharedObject::allocate(ComputeGraph* const graph) { in allocate()
45 void SharedObject::bind_users(ComputeGraph* const graph) { in bind_users()
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | Staging.h | 22 ComputeGraph& graph,
27 ComputeGraph& graph,
47 ComputeGraph& graph,
58 ComputeGraph& graph,
74 ComputeGraph& graph,
|
| D | UnaryOp.cpp | 24 ComputeGraph* graph, in resize_unary_op_node()
35 ComputeGraph& graph, in add_unary_op_node()
70 float get_val_or_inf(ComputeGraph& graph, const ValueRef& val, bool max) { in get_val_or_inf()
79 void op_name(ComputeGraph& graph, const std::vector<ValueRef>& args) { \
85 void op_name(ComputeGraph& graph, const std::vector<ValueRef>& args) { \
96 void op_name(ComputeGraph& graph, const std::vector<ValueRef>& args) { \
107 void op_name(ComputeGraph& graph, const std::vector<ValueRef>& args) { \
117 void gelu(ComputeGraph& graph, const std::vector<ValueRef>& args) { in gelu()
|
| D | Staging.cpp | 23 ComputeGraph& graph, in add_staging_to_tensor_node()
66 ComputeGraph& graph, in add_tensor_to_staging_node()
114 ComputeGraph& graph, in add_prepack_standard_node()
145 ComputeGraph& graph, in prepack_standard()
160 ComputeGraph& graph, in prepack_standard_like()
174 ComputeGraph& graph, in add_prepack_direct_copy_buffer_node()
199 ComputeGraph& graph, in prepack_direct_copy_buffer()
208 void prepack_op(ComputeGraph& graph, const std::vector<ValueRef>& args) { in prepack_op()
|
| D | Full.cpp | 19 ComputeGraph* graph, in resize_full_node()
35 ComputeGraph& graph, in add_full_node()
63 void full(ComputeGraph& graph, const std::vector<ValueRef>& args) { in full()
67 void zeros(ComputeGraph& graph, const std::vector<ValueRef>& args) { in zeros()
72 void ones(ComputeGraph& graph, const std::vector<ValueRef>& args) { in ones()
|
| D | Transpose.cpp | 22 ComputeGraph* graph, in resize_transpose_view_node()
39 ComputeGraph& graph, in check_transpose_view_args()
56 ComputeGraph& graph, in add_transpose_view_node()
73 void transpose(ComputeGraph& graph, const std::vector<ValueRef>& args) { in transpose()
|
| D | Clone.cpp | 22 ComputeGraph* graph, in resize_clone_node()
36 ComputeGraph& graph, in add_clone_node()
60 ComputeGraph& graph, in add_image_to_buffer_node()
84 ComputeGraph& graph, in add_buffer_to_image_node()
107 void clone(ComputeGraph& graph, const std::vector<ValueRef>& args) { in clone()
|
| D | Pool.cpp | 26 ComputeGraph* graph, in resize_pool2d_node()
66 ComputeGraph& graph, in add_max_pool2d_node()
116 void max_pool2d(ComputeGraph& graph, const std::vector<ValueRef>& args) { in max_pool2d()
131 ComputeGraph& graph, in create_divisor_params()
142 ComputeGraph& graph, in add_avg_pool2d_node()
192 void avg_pool2d(ComputeGraph& graph, const std::vector<ValueRef>& args) { in avg_pool2d()
|
| D | Arange.cpp | 20 ComputeGraph* graph, in resize_arange_node()
42 ComputeGraph& graph, in check_arange_input()
58 ComputeGraph& graph, in add_arange_node()
109 void arange(ComputeGraph& graph, const std::vector<ValueRef>& args) { in arange()
|
| D | MatMul.cpp | 22 const ComputeGraph& graph, in check_matmul_args()
38 ComputeGraph* graph, in resize_matmul_node()
60 ComputeGraph& graph, in add_matmul_naive_buffer_node()
111 ComputeGraph& graph, in add_matmul_naive_texture3d_node()
156 ComputeGraph& graph, in add_matmul_optimized_node()
242 ComputeGraph& graph, in add_matmul_node()
260 void matmul(ComputeGraph& graph, const std::vector<ValueRef>& args) { in matmul()
|
| D | ToCopy.cpp | 18 ComputeGraph* graph, in resize_to_copy_op_node()
28 void add_to_copy_node(ComputeGraph& graph, ValueRef in, ValueRef out) { in add_to_copy_node()
43 void to_copy(ComputeGraph& graph, const std::vector<ValueRef>& args) { in to_copy()
|
| D | Softmax.cpp | 21 ComputeGraph* graph, in resize_softmax_node()
33 ComputeGraph& graph, in add_softmax_node()
107 void softmax(ComputeGraph& graph, const std::vector<ValueRef>& args) { in softmax()
113 void log_softmax(ComputeGraph& graph, const std::vector<ValueRef>& args) { in log_softmax()
|
| D | Split.cpp | 21 ComputeGraph& graph, in add_split_with_sizes_default_node()
98 ComputeGraph& graph, in add_split_with_sizes_default_node()
110 ComputeGraph& graph, in split_with_sizes_copy_default()
116 ComputeGraph& graph, in add_split_tensor_node()
132 void split_tensor(ComputeGraph& graph, const std::vector<ValueRef>& args) { in split_tensor()
|
| D | Slice.cpp | 37 ComputeGraph& graph, in add_slice_tensor_copy_node()
155 ComputeGraph& graph, in get_slice_sizes()
180 ComputeGraph* graph, in resize_slice_view_node()
197 ComputeGraph& graph, in check_slice_view_args()
246 ComputeGraph& graph, in add_slice_view_node()
272 void slice_tensor_copy(ComputeGraph& graph, const std::vector<ValueRef>& args) { in slice_tensor_copy()
283 void slice_tensor(ComputeGraph& graph, const std::vector<ValueRef>& args) { in slice_tensor()
|
| /external/executorch/backends/vulkan/runtime/graph/ops/ |
| D | PrepackNode.h | 17 class ComputeGraph; variable
26 friend class ComputeGraph; variable
30 ComputeGraph& graph,
41 void encode(ComputeGraph* graph);
59 api::StagingBuffer create_staging_buffer(ComputeGraph* graph);
|
| D | ExecuteNode.h | 17 class ComputeGraph; variable
43 friend class ComputeGraph; variable
47 ComputeGraph*,
64 virtual void encode(ComputeGraph* graph) { in encode()
68 inline void trigger_resize(ComputeGraph* graph) { in trigger_resize()
|
| D | DispatchNode.h | 19 class ComputeGraph; variable
25 friend class ComputeGraph; variable
29 ComputeGraph& graph,
41 void encode(ComputeGraph* graph) override;
|
| D | BlitNode.h | 24 friend class ComputeGraph; variable
28 ComputeGraph& graph,
37 void encode(ComputeGraph* graph) override;
|
| D | PrepackNode.cpp | 19 vkapi::ShaderInfo get_noop_shader(ComputeGraph& graph, const ValueRef packed) { in get_noop_shader()
28 ComputeGraph& graph, in PrepackNode()
48 api::StagingBuffer PrepackNode::create_staging_buffer(ComputeGraph* graph) { in create_staging_buffer()
68 void PrepackNode::encode(ComputeGraph* graph) { in encode()
|
| /external/executorch/backends/vulkan/runtime/ |
| D | VulkanBackend.cpp | 151 ComputeGraph* compute_graph_;
159 ComputeGraph* compute_graph, in GraphBuilder()
390 ComputeGraph* graph, in maybe_resize_input()
425 ComputeGraph* graph, in maybe_update_scalar_tensor()
445 ComputeGraph* graph, in maybe_resize_output()
478 compileModel(const void* buffer_pointer, ComputeGraph* compute_graph) const { in compileModel()
523 ComputeGraph* compute_graph = ET_ALLOCATE_INSTANCE_OR_RETURN_ERROR( in init()
524 context.get_runtime_allocator(), ComputeGraph); in init()
526 new (compute_graph) ComputeGraph(get_graph_config(compile_specs)); in init()
547 ComputeGraph* compute_graph = static_cast<ComputeGraph*>(handle); in execute()
[all …]
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/ |
| D | KernelUtils.h | 36 ComputeGraph& graph,
44 ComputeGraph& graph,
58 ComputeGraph& graph,
|
| D | KernelUtils.cpp | 13 utils::ivec2 make_ivec2_from_list(ComputeGraph& graph, ValueRef vref) { in make_ivec2_from_list()
18 ComputeGraph& graph, in make_ivec2_kernel_size()
30 ComputeGraph& graph, in create_kernel2d_params()
45 ComputeGraph& graph, in create_kernel2d_params()
149 ComputeGraph& graph, in calc_out_sizes_hw()
|