Searched refs:reference_tensor (Results 1 – 3 of 3) sorted by relevance
412 const Tensor& reference_tensor, in broadcast_to_outnames() argument414 if (!tensor.has_names() && !reference_tensor.has_names()) { in broadcast_to_outnames()417 auto reference_names = reference_tensor.names(); in broadcast_to_outnames()
130 const Tensor& reference_tensor,
168 const at::Tensor reference_tensor = tensors_to_verify[i]; in verify_tensors() local171 if (!rank_tensor.equal(reference_tensor)) { in verify_tensors()