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.h37 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
104 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dmean_stddev_normalization.h37 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dsoftmax1x1.h32 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dconvolution_transposed_4x4.h43 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dconv_metal.h60 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
Dfully_connected.h122 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dworkgroup_selection.cc36 std::vector<T>* work_groups) { in AddCornerCases()
150 std::vector<T> work_groups; in GenerateWorkGroupSizes() local
193 std::vector<T>* work_groups) { in GenerateWorkGroupSizesAlignedToGrid()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h98 std::vector<int3>* work_groups) const override { in GetPossibleKernelWorkGroups()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc56 std::vector<int3> work_groups; in DispatchKernel() local