Searched refs:SelectPooling (Results 1 – 3 of 3) sorted by relevance
40 std::unique_ptr<GPUOperation> SelectPooling(const Pooling2DAttributes& attr,
65 std::unique_ptr<GPUOperation> SelectPooling(const Pooling2DAttributes& attr, in SelectPooling() function
450 *gpu_op = SelectPooling(attr, op_def); in GPUOperationFromNode()