Home
last modified time | relevance | path

Searched refs:SelectReduce (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.h72 std::unique_ptr<GPUOperation> SelectReduce(const std::set<Axis>& axis_to_reduce,
Dsimple_selectors.cc161 std::unique_ptr<GPUOperation> SelectReduce(const std::set<Axis>& axis_to_reduce, in SelectReduce() function
Doperation_selector.cc431 *gpu_op = SelectReduce(attr.dims, inputs[0]->tensor.shape, op_type, in GPUOperationFromNode()
558 *gpu_op = SelectReduce(attr.dims, inputs[0]->tensor.shape, op_type, in GPUOperationFromNode()