Searched refs:reduction_indices_vector (Results 1 – 2 of 2) sorted by relevance
2782 const TensorVector& reduction_indices_vector) const { in IsReductionSimplifiableToIdentity()2783 int output_size = reduction_indices_vector[0]->NumElements(); in IsReductionSimplifiableToIdentity()2794 if (reduction_indices_vector[0]->dtype() == DT_INT32) { in IsReductionSimplifiableToIdentity()2795 dim = reduction_indices_vector[0]->flat<int32>()(i); in IsReductionSimplifiableToIdentity()2797 dim = reduction_indices_vector[0]->flat<int64>()(i); in IsReductionSimplifiableToIdentity()2850 TensorVector reduction_indices_vector; in SimplifyReduction() local2851 auto outputs_cleanup = gtl::MakeCleanup([&reduction_indices_vector] { in SimplifyReduction()2852 for (const auto& out : reduction_indices_vector) { in SimplifyReduction()2857 &reduction_indices_vector) in SimplifyReduction()2859 reduction_indices_vector.size() != 1) { in SimplifyReduction()[all …]
238 const gtl::InlinedVector<TensorValue, 4>& reduction_indices_vector) const;