Searched refs:accumulator_j_shape (Results 1 – 1 of 1) sorted by relevance
377 shape_inference::ShapeHandle accumulator_j_shape; in operator ()() local378 TF_RETURN_IF_ERROR(c->WithRank(c->input(j), 2, &accumulator_j_shape)); in operator ()()380 TF_RETURN_IF_ERROR(c->Merge(parameter_shape, accumulator_j_shape, &merged)); in operator ()()