ONE - On-device Neural Engine
Loading...
Searching...
No Matches
arm_compute::CLBinaryLogicalOp Class Reference

#include <CLBinaryLogicalOp.h>

Collaboration diagram for arm_compute::CLBinaryLogicalOp:

Public Member Functions

void configure (ICLTensor *input1, ICLTensor *input2, ICLTensor *output, BinaryLogicalOperation op)
 

Detailed Description

Definition at line 52 of file CLBinaryLogicalOp.h.

Member Function Documentation

◆ configure()

void CLBinaryLogicalOp::configure ( ICLTensor *  input1,
ICLTensor *  input2,
ICLTensor *  output,
BinaryLogicalOperation  op 
)

Initialise the function's source and destination.

Parameters
[in]input1Source tensor1. Data types supported: U8, QASYMM8.
[in]input2Source tensor2. Data types supported: U8 QASYMM8.
[out]outputOutput tensor. Data types supported: U8, QASYMM8.

Definition at line 49 of file CLBinaryLogicalOp.cpp.

51{
52 auto k = std::make_unique<CLBinaryLogicalOpKernel>();
53 k->configure(input1, input2, output, op);
54 _kernel = std::move(k);
55
56 if (output->info()->dimension(0) > 1)
57 {
58 ICLTensor *broadcasted_info = (input1->info()->dimension(0) == 1) ? input1 : input2;
59 if (broadcasted_info->info()->dimension(0) == 1)
60 {
61 _border_handler->configure(broadcasted_info, _kernel->border_size(), BorderMode::REPLICATE);
62 }
63 }
64}

The documentation for this class was generated from the following files: