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.cc449 conv_params.block_size.w * 4 * conv_params.src_depth_loop_size; in GenerateConv()
977 for (int i = 1; i < conv_params.src_depth_loop_size; ++i) { in GenerateConv()
1107 conv_params.src_depth_loop_size = 1; in GuessBestParams()
1135 conv_params.src_depth_loop_size = 2; in GuessBestParams()
1138 conv_params.src_depth_loop_size = 4; in GuessBestParams()
1155 conv_params.src_depth_loop_size = 1; in GuessBestParams()
1170 conv_params.src_depth_loop_size = 2; in GuessBestParams()
1173 conv_params.src_depth_loop_size = 4; in GuessBestParams()
1177 conv_params.src_depth_loop_size = 2; in GuessBestParams()
1180 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