Home
last modified time | relevance | path

Searched refs:need_dst_loop (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_metal.cc204 !params.need_dst_loop && !params.need_src_loop && params.x_kernel_is_1 && in GenerateConvolution()
265 params.need_dst_loop ? " + Z * 4 * args.src_tensor.Slices()" : ""; in GenerateConvolution()
266 if (!params.need_dst_loop) { in GenerateConvolution()
534 if (params.need_dst_loop) { in GenerateConvolution()
758 params.need_dst_loop = false; in GetConvParamsForA7A8()
761 !params.need_dst_loop && !params.need_src_loop && params.x_kernel_is_1 && in GetConvParamsForA7A8()
841 params.need_dst_loop = false; in GetConvParamsForA9AndHigher()
844 !params.need_dst_loop && !params.need_src_loop && params.x_kernel_is_1 && in GetConvParamsForA9AndHigher()
906 params.need_dst_loop = true; in GetConvParamsForAMD()
942 params.need_dst_loop = true; in GetConvParams()
[all …]
Dconv_metal.h46 bool need_dst_loop = true; member