Searched refs:max_tensor_display_size (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/jit/ir/ | ||
| D | attributes.h | 15 constexpr int max_tensor_display_size = 10; variable |
| D | ir.cpp | 170 } else if (tensor.numel() <= max_tensor_display_size) { in printAttribute() |