Searched defs:pairwise_distance (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | distance.h | 46 inline Tensor pairwise_distance( in pairwise_distance() function
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Distance.cpp | 48 Tensor pairwise_distance(const Tensor& x1, const Tensor& x2, double p, double eps, bool keepdim) { in pairwise_distance() function
|
| /external/pytorch/torch/_refs/nn/functional/ |
| D | __init__.py | 1187 def pairwise_distance( function
|
| /external/pytorch/torch/nn/ |
| D | functional.py | 5105 pairwise_distance = _add_docstr( variable
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset9.py | 3055 def pairwise_distance(g: jit_utils.GraphContext, input1, input2, p, eps, keepdim): function
|
| /external/pytorch/test/cpp/api/ |
| D | modules.cpp | 2201 auto pairwise_distance = [&](const torch::Tensor& x, const torch::Tensor& y) { in TEST_F() local
|