Searched refs:num_dims_output (Results 1 – 1 of 1) sorted by relevance
449 int num_dims_output = NumDimensions(output); in Eval() local451 TF_LITE_ENSURE_EQ(context, num_dims_output, NumDimensions(input)); in Eval()452 TF_LITE_ENSURE(context, num_dims_output >= 2); in Eval()453 TF_LITE_ENSURE_EQ(context, output_shape.Dims(num_dims_output - 2), in Eval()455 TF_LITE_ENSURE_EQ(context, output_shape.Dims(num_dims_output - 1), in Eval()