Function _convert_matrix_set_diag

tensorflow/python/ops/parallel_for/pfor.py:1806–1810 · view source on GitHub ↗

(pfor_input)

Source from the content-addressed store, hash-verified

1804
1805	@RegisterPFor("MatrixSetDiag")
1806	def _convert_matrix_set_diag(pfor_input):
1807	pfor_input.stack_inputs()
1808	t = pfor_input.stacked_input(0)
1809	diag = pfor_input.stacked_input(1)
1810	return wrap(array_ops.matrix_set_diag(t, diag), True)
1811
1812
1813	# Registrations for MatrixDiagV2, MatrixDiagPartv2, and MatrixSetDiagV2.

nothing calls this directly

stack_inputsMethod · 0.80

stacked_inputMethod · 0.80

wrapFunction · 0.70

no test coverage detected