Searched refs:method_output_tensor (Results 1 – 1 of 1) sorted by relevance
150 const Tensor& method_output_tensor, in tensors_are_close() argument153 if (bundled_tensor.scalar_type() != method_output_tensor.scalar_type() || in tensors_are_close()154 bundled_tensor.sizes() != method_output_tensor.sizes()) { in tensors_are_close()161 bundled_tensor.strides() == method_output_tensor.strides(), in tensors_are_close()180 method_output_tensor.strides(), in tensors_are_close()200 method_output_tensor.const_data_ptr<float>(), in tensors_are_close()207 method_output_tensor.const_data_ptr<double>(), in tensors_are_close()214 method_output_tensor.const_data_ptr<Half>(), in tensors_are_close()222 method_output_tensor.const_data_ptr(), in tensors_are_close()374 const auto method_output_tensor = method_output.toTensor(); in verify_method_outputs() local[all …]