Searched defs:outer_shape (Results 1 – 2 of 2) sorted by relevance
53 std::vector<unsigned int> outer_shape; in InferOutputShapes() local
663 Output outer_shape, flat_values_shape; in BatchGatherGrad() local727 auto outer_shape = Slice(scope, params_shape, {0}, axis_expand); in GatherV2Grad() local