MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / _convert_matrix_band_part

Function _convert_matrix_band_part

tensorflow/python/ops/parallel_for/pfor.py:1797–1802  ·  view source on GitHub ↗
(pfor_input)

Source from the content-addressed store, hash-verified

1795
1796@RegisterPFor("MatrixBandPart")
1797def _convert_matrix_band_part(pfor_input):
1798 t = pfor_input.stacked_input(0)
1799 num_lower = pfor_input.unstacked_input(1)
1800 num_upper = pfor_input.unstacked_input(2)
1801 return wrap(array_ops.matrix_band_part(
1802 t, num_lower=num_lower, num_upper=num_upper), True)
1803
1804
1805@RegisterPFor("MatrixSetDiag")

Callers

nothing calls this directly

Calls 3

stacked_inputMethod · 0.80
unstacked_inputMethod · 0.80
wrapFunction · 0.70

Tested by

no test coverage detected