Searched refs:hashed_layout (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
H A D | Embedding.cpp | 53 {t_out->hashed_layout(), in add_embedding_node() 54 t_in->hashed_layout(), in add_embedding_node() 55 t_weight->hashed_layout()})); in add_embedding_node()
|
H A D | BinaryOp.cpp | 92 {t_out->hashed_layout(), t_in1->hashed_layout(), t_in2->hashed_layout()}, in add_binary_op_node()
|
H A D | Convolution.cpp | 111 {t->hashed_layout()})); in prepack_biases() 486 {t_out->hashed_layout(), in add_conv1d_node() 487 t_in->hashed_layout(), in add_conv1d_node() 488 t_weight->hashed_layout(), in add_conv1d_node() 489 t_bias->hashed_layout()}, in add_conv1d_node()
|
H A D | NativeLayerNorm.cpp | 116 t_input->hashed_layout(), in add_native_layer_norm_node() 117 t_out->hashed_layout(), in add_native_layer_norm_node()
|
/aosp_15_r20/external/executorch/backends/vulkan/test/utils/ |
H A D | test_utils.cpp | 73 vkapi::SpecVarList specialization_constants = {v_dst.hashed_layout()}; in record_nchw_to_image_op() 97 vkapi::SpecVarList specialization_constants = {v_src.hashed_layout()}; in record_image_to_nchw_op() 129 {v_src.hashed_layout()}, in record_bitw8_image_to_nchw_nobitw8buffer_op() 337 {out.hashed_layout(), mat1.hashed_layout(), mat2.hashed_layout()}, in record_matmul_texture3d()
|
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/ |
H A D | Tensor.h | 440 inline int32_t hashed_layout() const { in hashed_layout() function
|
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ |
H A D | ComputeGraph.h | 322 return values_.at(idx).toConstTensor().hashed_layout();
|