Searched refs:set_output_tensor (Results 1 – 10 of 10) sorted by relevance
1229 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 …]
129 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()
182 ValueRef staging_out = graph.set_output_tensor(r_out); in test_vulkan_linear_int4()
366 ValueRef staging_out = graph.set_output_tensor(r_out); in test_vulkan_sdpa()
144 void set_output_tensor(TensorID tid, size_t idx);
77 void INode::set_output_tensor(TensorID tid, size_t idx) in set_output_tensor() function in arm_compute::graph::INode
88 node->set_output_tensor(new_output->id(), 0); in set_new_output_and_inherit_accessor()
546 ValueRef set_output_tensor(const ValueRef idx, const bool use_staging = true);
399 ValueRef ComputeGraph::set_output_tensor( in set_output_tensor() function in vkcompute::ComputeGraph
379 compute_graph_->set_output_tensor(ref); in build_graph()