Home
last modified time | relevance | path

Searched refs:set_output_tensor (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dvulkan_compute_api_test.cpp1229 out.staging = graph.set_output_tensor(out.value); in TEST()
1272 out.staging = graph.set_output_tensor(out.value); in TEST()
1329 out.staging = graph.set_output_tensor(out.value); in TEST()
1414 out.staging = graph.set_output_tensor(out.value); in TEST()
1471 out.staging = graph.set_output_tensor(out.value); in TEST()
1539 out.staging = graph.set_output_tensor(out.value); in TEST()
1634 out.staging = graph.set_output_tensor(out.value); in TEST()
1780 out.staging = graph.set_output_tensor(out.value); in TEST()
1846 out.staging = graph.set_output_tensor(out.value); in TEST()
1923 out.staging = graph.set_output_tensor(out.value); in test_clone()
[all …]
/aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/
H A Drotary_embedding_test.cpp129 ValueRef staging_xq_out = graph.set_output_tensor(r_xq_out); in test_reference()
130 ValueRef staging_xk_out = graph.set_output_tensor(r_xk_out); in test_reference()
H A Dlinear_weight_int4_test.cpp182 ValueRef staging_out = graph.set_output_tensor(r_out); in test_vulkan_linear_int4()
H A Dsdpa_test.cpp366 ValueRef staging_out = graph.set_output_tensor(r_out); in test_vulkan_sdpa()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/
H A DINode.h144 void set_output_tensor(TensorID tid, size_t idx);
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DINode.cpp77 void INode::set_output_tensor(TensorID tid, size_t idx) in set_output_tensor() function in arm_compute::graph::INode
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DInPlaceOperationMutator.cpp88 node->set_output_tensor(new_output->id(), 0); in set_new_output_and_inherit_accessor()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/
H A DComputeGraph.h546 ValueRef set_output_tensor(const ValueRef idx, const bool use_staging = true);
H A DComputeGraph.cpp399 ValueRef ComputeGraph::set_output_tensor( in set_output_tensor() function in vkcompute::ComputeGraph
/aosp_15_r20/external/executorch/backends/vulkan/runtime/
H A DVulkanBackend.cpp379 compute_graph_->set_output_tensor(ref); in build_graph()