Searched refs:raw_dim_axis (Results 1 – 1 of 1) sorted by relevance
43 DimensionHandle raw_dim_axis; in ReduceSliceShapeFn() local44 TF_RETURN_IF_ERROR(c->Max(c->Dim(c->input(1), 0), 1, &raw_dim_axis)); in ReduceSliceShapeFn()45 TF_RETURN_IF_ERROR(c->Subtract(raw_dim_axis, 1, &dim_axis)); in ReduceSliceShapeFn()