Searched refs:elementwise_ (Results 1 – 8 of 8) sorted by relevance
98 elementwise_(operation.elementwise_), in GPUOperation()120 elementwise_ = operation.elementwise_; in operator =()187 if (elementwise_) { in AssembleCode()217 if (elementwise_ || tensor_to_grid_ == TensorToGrid::kWBToX_HDToY_SToZ) { in GetGridSize()
135 bool IsLinkable() const { return elementwise_ && linkable_; } in IsLinkable()147 bool elementwise_ = false; variable
188 op.elementwise_ = true; in CreateElementwiseOneRuntimeOneScalar()215 result.elementwise_ = true; in CreateElementwiseTwoInput()247 result.elementwise_ = true; in CreateElementwiseTwoInput()272 op.elementwise_ = true; in CreateElementwiseOneInput()307 op.elementwise_ = true; in CreateElementwiseTwoInput()
26 op.elementwise_ = true; in CreateReLU()
31 add.elementwise_ = true; in CreateAdd()
41 op.elementwise_ = true; in CreateQuantizeAndDequantize()
30 result.elementwise_ = true; in CreatePReLU()
816 op->elementwise_ = fb_op->elementwise(); in Decode()883 op_builder.add_elementwise(op.elementwise_); in Encode()