Searched refs:reduction_indices_t (Results 1 – 1 of 1) sorted by relevance
1132 Status ReductionShapeHelper(const Tensor* reduction_indices_t, in ReductionShapeHelper() argument1135 auto reduction_indices = reduction_indices_t->flat<T>(); in ReductionShapeHelper()1136 for (int i = 0; i < reduction_indices_t->NumElements(); ++i) { in ReductionShapeHelper()1169 const Tensor* reduction_indices_t = c->input_tensor(1); in ReductionShape() local1170 if (reduction_indices_t == nullptr || !c->RankKnown(input)) { in ReductionShape()1185 if (reduction_indices_t->dtype() == DataType::DT_INT32) { in ReductionShape()1186 TF_RETURN_IF_ERROR(ReductionShapeHelper<int32>(reduction_indices_t, in ReductionShape()1188 } else if (reduction_indices_t->dtype() == DataType::DT_INT64) { in ReductionShape()1189 TF_RETURN_IF_ERROR(ReductionShapeHelper<int64>(reduction_indices_t, in ReductionShape()