Searched refs:dim0_value (Results 1 – 1 of 1) sorted by relevance
1049 auto dim0_value = dim0->scalar<int32>()(); in SparseSegmentReductionGradShapeFn() local1050 if (dim0_value < 0) { in SparseSegmentReductionGradShapeFn()1054 dim0_shape = c->Vector(dim0_value); in SparseSegmentReductionGradShapeFn()1091 auto dim0_value = dim0->scalar<int32>()(); in SparseSegmentReductionWithNumSegmentsShapeFn() local1092 if (dim0_value < 0) { in SparseSegmentReductionWithNumSegmentsShapeFn()1096 TF_RETURN_IF_ERROR(c->Concatenate(c->Vector(dim0_value), subshape, &out)); in SparseSegmentReductionWithNumSegmentsShapeFn()