Searched refs:reduction_indices_t (Results 1 – 1 of 1) sorted by relevance
1806 Status ReductionShapeHelper(const Tensor* reduction_indices_t, in ReductionShapeHelper() argument1809 auto reduction_indices = reduction_indices_t->flat<T>(); in ReductionShapeHelper()1810 for (int i = 0; i < reduction_indices_t->NumElements(); ++i) { in ReductionShapeHelper()1843 const Tensor* reduction_indices_t = c->input_tensor(1); in ReductionShape() local1844 if (reduction_indices_t == nullptr || !c->RankKnown(input)) { in ReductionShape()1859 if (reduction_indices_t->dtype() == DataType::DT_INT32) { in ReductionShape()1860 TF_RETURN_IF_ERROR(ReductionShapeHelper<int32>(reduction_indices_t, in ReductionShape()1862 } else if (reduction_indices_t->dtype() == DataType::DT_INT64) { in ReductionShape()1863 TF_RETURN_IF_ERROR(ReductionShapeHelper<int64>(reduction_indices_t, in ReductionShape()