Searched refs:_print_tensor (Results 1 – 1 of 1) sorted by relevance
890 def _print_tensor(tensor_name, num_elements, tensor, output_tensor): function932 return _print_tensor(tensor_name, _TRACE_MODE_PART_TENSOR_SIZE,938 return _print_tensor(tensor_name, -1, tensor, tensor)