MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / _add_with_axis

Function _add_with_axis

python/paddle/tensor/math.py:1401–1407  ·  view source on GitHub ↗
(x, y, axis=-1, name=None)

Source from the content-addressed store, hash-verified

1399
1400
1401def _add_with_axis(x, y, axis=-1, name=None):
1402 # opt performance, only dynamic mode needs reshape
1403 if in_dynamic_or_pir_mode():
1404 return _elementwise_op_with_axis(x, y, axis, name, "add")
1405 else:
1406 op_type = 'elementwise_add'
1407 return _elementwise_op(LayerHelper(op_type, **locals()))
1408
1409
1410def _subtract_with_axis(x, y, axis=-1, name=None):

Callers 6

deform_conv2dFunction · 0.90
_conv_ndFunction · 0.90
conv1dFunction · 0.90
conv1d_transposeFunction · 0.90
conv2d_transposeFunction · 0.90
conv3d_transposeFunction · 0.90

Calls 4

in_dynamic_or_pir_modeFunction · 0.85
_elementwise_opFunction · 0.85
LayerHelperClass · 0.85

Tested by

no test coverage detected