Searched refs:set_input_tensor (Results 1 – 5 of 5) sorted by relevance
545 ValueRef set_input_tensor(const ValueRef idx, const bool use_staging = true);583 ValueRef staging = set_input_tensor(t);597 ValueRef staging = set_input_tensor(t);611 ValueRef staging = set_input_tensor(t);627 ValueRef staging = set_input_tensor(t);
382 ValueRef ComputeGraph::set_input_tensor( in set_input_tensor() function in vkcompute::ComputeGraph
330 compute_graph_->set_input_tensor(ref); in build_graph()
442 "def set_input_tensor(interpreter, image):\n",459 " set_input_tensor(interpreter, image)\n",
2558 ValueRef r_staging_in = graph.set_input_tensor(r_tensor); in compute_graph_round_trip_test()2776 mat2.staging = graph.set_input_tensor(mat2.value); in test_mm()