Searched refs:reduction_indices_vector (Results 1 – 2 of 2) sorted by relevance
2512 const TensorVector& reduction_indices_vector) const { in IsReductionSimplifiableToIdentity()2513 int output_size = reduction_indices_vector[0]->NumElements(); in IsReductionSimplifiableToIdentity()2524 if (reduction_indices_vector[0]->dtype() == DT_INT32) { in IsReductionSimplifiableToIdentity()2525 dim = reduction_indices_vector[0]->flat<int32>()(i); in IsReductionSimplifiableToIdentity()2527 dim = reduction_indices_vector[0]->flat<int64>()(i); in IsReductionSimplifiableToIdentity()2555 TensorVector reduction_indices_vector; in SimplifyReduction() local2556 auto outputs_cleanup = gtl::MakeCleanup([&reduction_indices_vector] { in SimplifyReduction()2557 for (const auto& out : reduction_indices_vector) { in SimplifyReduction()2562 &reduction_indices_vector) in SimplifyReduction()2564 reduction_indices_vector.size() != 1) { in SimplifyReduction()[all …]
166 const gtl::InlinedVector<TensorValue, 4>& reduction_indices_vector) const;