Searched refs:dim0_shape (Results 1 – 1 of 1) sorted by relevance
1043 ShapeHandle dim0_shape; in SparseSegmentReductionGradShapeFn() local1047 dim0_shape = c->Vector(InferenceContext::kUnknownDim); in SparseSegmentReductionGradShapeFn()1054 dim0_shape = c->Vector(dim0_value); in SparseSegmentReductionGradShapeFn()1058 TF_RETURN_IF_ERROR(c->Concatenate(dim0_shape, subshape, &out)); in SparseSegmentReductionGradShapeFn()