Home
last modified time | relevance | path

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 Dxnn_compiler.cpp23 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 DXNNCompiler.cpp164 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 DVulkanBackend.cpp67 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()