Searched refs:flatbuffer_graph (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/compiler/ |
| H A D | xnn_compiler.cpp | 23 auto flatbuffer_graph = fb_xnnpack::GetXNNGraph(buffer_pointer); in compileModel() local
31 /*external_value_ids=*/flatbuffer_graph->num_externs(), in compileModel()
43 for (auto value : *flatbuffer_graph->xvalues()) { in compileModel()
54 const auto& constant_buffer = *flatbuffer_graph->constant_buffer(); in compileModel()
81 for (auto node : *flatbuffer_graph->xnodes()) { in compileModel()
109 for (auto old_id : *flatbuffer_graph->input_ids()) { in compileModel()
113 for (auto old_id : *flatbuffer_graph->output_ids()) { in compileModel()
|
| /aosp_15_r20/external/executorch/backends/xnnpack/runtime/ |
| H A D | XNNCompiler.cpp | 164 GraphPtr flatbuffer_graph, in getConstantDataPtr() argument
171 const auto& constant_buffer = *flatbuffer_graph->constant_buffer(); in getConstantDataPtr()
174 const auto& constant_data_offsets = *flatbuffer_graph->constant_data(); in getConstantDataPtr()
193 GraphPtr flatbuffer_graph, in defineTensor() argument
235 getConstantDataPtr(tensor_value, flatbuffer_graph, constant_data_ptr); in defineTensor()
694 const fb_xnnpack::XNNGraph* flatbuffer_graph) noexcept { in defineConvertNode() argument
695 MAYBE_UNUSED(flatbuffer_graph); in defineConvertNode()
703 if (isQP8(flatbuffer_graph, node)) { in defineConvertNode()
1886 auto flatbuffer_graph = fb_xnnpack::GetXNNGraph(flatbuffer_data); in compileModel() local
1898 /*external_value_ids=*/flatbuffer_graph->num_externs(), in compileModel()
[all …]
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/ |
| H A D | VulkanBackend.cpp | 67 VkGraphPtr flatbuffer_graph, in get_constant_data_ptr() argument
70 VkBytesPtr constant_bytes = flatbuffer_graph->constants()->Get(buffer_idx); in get_constant_data_ptr()
502 VkGraphPtr flatbuffer_graph = vkgraph::GetVkGraph(flatbuffer_data); in compileModel() local
505 GraphBuilder(compute_graph, flatbuffer_graph, constant_data); in compileModel()
|