Searched defs:check_tensor (Results 1 – 6 of 6) sorted by relevance
114 def check_tensor(shards_metadata, tensor_dims) -> None: function
54 void check_tensor(const TensorInfo& tensor_info) { in check_tensor() function
53 at::Tensor& check_tensor) { in assert_inf_and_nan()
103 def check_tensor(sample): function
292 static inline void check_tensor( in check_tensor() function
193 void check_tensor(const std::vector<at::Tensor>& tensors) { in check_tensor() function