Searched refs:input_distance_mat (Results 1 – 1 of 1) sorted by relevance
43 const Tensor& input_distance_mat = context->input(0); in Compute() local44 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()