Searched refs:record_scalar_tensor (Results 1 – 1 of 1) sorted by relevance
325 def record_scalar_tensor(x, prefix): function343 torch.ops.mylib.record_scalar_tensor.default, _EffectType.ORDERED383 torch.ops.mylib.record_scalar_tensor(output.mean(), prefix)385 torch.ops.mylib.record_scalar_tensor(output.max(), prefix)388 torch.ops.mylib.record_scalar_tensor(output.sum(), prefix)