Home
last modified time | relevance | path

Searched defs:work_groups (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dwork_group_picking.cc38 std::vector<int3> work_groups; in GenerateWorkGroupSizesXYMultipleOf() local
68 std::vector<int3> work_groups; in GenerateWorkGroupSizesXMultipleOf() local
94 std::vector<int3>* work_groups) { in GetWorkGroupsAlignedToGrid()
241 std::vector<int3>* work_groups) { in GetPossibleWorkGroupsXYMultipleOf()
250 std::vector<int3>* work_groups) { in GetPossibleWorkGroupsXMultipleOf()
273 std::vector<int3>* work_groups) { in GetPossibleWorkGroups()
295 std::vector<int3>* work_groups) { in GetPossibleWorkGroupsConv()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dwinograd.h38 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
108 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dmean_stddev_normalization.h43 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dsoftmax1x1.h34 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dconvolution_transposed_4x4.h43 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dconv_metal_simd.h35 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dfully_connected.h123 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dworkgroup_selection.cc37 std::vector<T>* work_groups) { in AddCornerCases()
151 std::vector<T> work_groups; in GenerateWorkGroupSizes() local
194 std::vector<T>* work_groups) { in GenerateWorkGroupSizesAlignedToGrid()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h102 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc60 std::vector<int3> work_groups; in DispatchKernel() local