Searched refs:distance_mat (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| D | image_ops_test.py | 294 def _BipartiteMatchTest(self, distance_mat, distance_mat_shape, argument
298 distance_mat_np = np.array(distance_mat, dtype=np.float32).reshape(
316 distance_mat = [0.5, 0.8, 0.1,
321 self._BipartiteMatchTest(distance_mat, [2, 3], num_valid_rows,
329 self._BipartiteMatchTest(distance_mat, [2, 3], num_valid_rows,
337 self._BipartiteMatchTest(distance_mat, [2, 3], num_valid_rows,
346 self._BipartiteMatchTest(distance_mat, [2, 3], num_valid_rows,
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| D | bipartite_match_op.cc | 88 TTypes<float, 2>::ConstTensor distance_mat = in Compute() local
98 DistancePair(index1, index2, distance_mat(index1, index2))); in Compute()
|
| /external/tensorflow/tensorflow/contrib/image/python/ops/ |
| D | image_ops.py | 431 def bipartite_match(distance_mat, argument
465 distance_mat, num_valid_rows, top_k, name=name)
|