hello khang.l4es,
according to Topic 191446, TC358840 should works on L4T 32.5.1 for Xavier NX.
it should be the regulator issue, please disassembler the dtb file into text file to review your device tree settings,
for example, $ dtc -I dtb -O dts -o output.txt tegra194-p3668-all-p3509-0000.dtb
thanks