Home
last modified time | relevance | path

Searched refs:input_distance_mat (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/image/kernels/
Dbipartite_match_op.cc43 const Tensor& input_distance_mat = context->input(0); in Compute() local
44 OP_REQUIRES(context, input_distance_mat.dims() == 2, in Compute()
47 input_distance_mat.shape().DebugString())); in Compute()
48 const int num_input_rows = input_distance_mat.dim_size(0); in Compute()
49 const int num_input_columns = input_distance_mat.dim_size(1); in Compute()
89 input_distance_mat.shaped<float, 2>( in Compute()