Searched defs:outer_shape (Results 1 – 1 of 1) sorted by relevance
662 Output outer_shape, flat_values_shape; in BatchGatherGrad() local726 auto outer_shape = Slice(scope, params_shape, {0}, axis_expand); in GatherV2Grad() local