Home
last modified time | relevance | path

Searched refs:src_depth_loop_size (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_powervr.cc446 conv_params.block_size.w * 4 * conv_params.src_depth_loop_size; in GenerateConv()
958 for (int i = 1; i < conv_params.src_depth_loop_size; ++i) { in GenerateConv()
1083 conv_params.src_depth_loop_size = 1; in GuessBestParams()
1111 conv_params.src_depth_loop_size = 2; in GuessBestParams()
1114 conv_params.src_depth_loop_size = 4; in GuessBestParams()
1131 conv_params.src_depth_loop_size = 1; in GuessBestParams()
1146 conv_params.src_depth_loop_size = 2; in GuessBestParams()
1149 conv_params.src_depth_loop_size = 4; in GuessBestParams()
1153 conv_params.src_depth_loop_size = 2; in GuessBestParams()
1156 conv_params.src_depth_loop_size = 4; in GuessBestParams()
[all …]
Dconv_metal.cc182 params.block_size.z * 4 * params.src_depth_loop_size; in GenerateConvolution()
498 for (int i = 1; i < params.src_depth_loop_size; ++i) { in GenerateConvolution()
505 std::to_string(params.block_size.z * 4 * params.src_depth_loop_size) + in GenerateConvolution()
702 params.src_depth_loop_size = 1; in GetConvParamsForA7A8()
754 if (params.src_depth_loop_size == src_slices) { in GetConvParamsForA7A8()
802 params.src_depth_loop_size = 1; in GetConvParamsForA9AndHigher()
828 params.src_depth_loop_size = 4; in GetConvParamsForA9AndHigher()
830 params.src_depth_loop_size = 2; in GetConvParamsForA9AndHigher()
834 params.src_depth_loop_size = 2; in GetConvParamsForA9AndHigher()
837 if (params.src_depth_loop_size == src_slices) { in GetConvParamsForA9AndHigher()
[all …]
Dconv_metal.h44 int src_depth_loop_size; member
Dconv_powervr.h90 int src_depth_loop_size; member