Searched refs:add_input_tensor (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | vulkan_compute_api_test.cpp | 1219 IOValueRef a = graph.add_input_tensor(size_big, vkapi::kFloat); in TEST()
1220 IOValueRef b = graph.add_input_tensor(size_small, vkapi::kFloat); in TEST()
1263 IOValueRef a = graph.add_input_tensor(sizes, vkapi::kFloat, utils::kBuffer); in TEST()
1316 IOValueRef orig = graph.add_input_tensor(orig_sizes, vkapi::kFloat); in TEST()
1404 IOValueRef a = graph.add_input_tensor(size_big, vkapi::kFloat); in TEST()
1405 IOValueRef b = graph.add_input_tensor(size_small, vkapi::kFloat); in TEST()
1450 IOValueRef a = graph.add_input_tensor(sizes, vkapi::kFloat); in TEST()
1519 IOValueRef a = graph.add_input_tensor(size_big, vkapi::kFloat); in TEST()
1582 IOValueRef a = graph.add_input_tensor( in TEST()
1586 IOValueRef b = graph.add_input_tensor( in TEST()
[all …]
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/ |
| H A D | etnaviv_ml.c | 104 … i, "NN", operation->input_tensor, operation->output_tensor, operation->add_input_tensor); in dump_graph()
158 if (operation->add_input_tensor == input_tensor) in lower_operations()
159 operation->add_input_tensor = output_tensor; in lower_operations()
171 operation->add_input_tensor, in lower_operations()
|
| H A D | etnaviv_ml.h | 62 unsigned add_input_tensor; member
|
| H A D | etnaviv_ml_nn.c | 477 operation->add_input_tensor = poperation->add.input_tensor->index; in etna_ml_lower_add()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ |
| H A D | ComputeGraph.h | 578 inline IOValueRef add_input_tensor(
591 inline IOValueRef add_input_tensor(
605 inline IOValueRef add_input_tensor(
619 inline IOValueRef add_input_tensor(
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | rotary_embedding_test.cpp | 108 IOValueRef r_##x = graph.add_input_tensor( \ in test_reference()
|
| H A D | linear_weight_int4_test.cpp | 165 IOValueRef r_##x = graph.add_input_tensor( \ in test_vulkan_linear_int4()
|
| H A D | sdpa_test.cpp | 337 IOValueRef r_##x = graph.add_input_tensor( \ in test_vulkan_sdpa()
|