Searched refs:GetSpatialDimsToSplit (Results 1 – 1 of 1) sorted by relevance
84 std::pair<std::vector<int64_t>, std::vector<int64_t>> GetSpatialDimsToSplit(356 ConvolutionVisitor::GetSpatialDimsToSplit(HloInstruction* old_operand) { in GetSpatialDimsToSplit() function in xla::__anon5e1d62920111::ConvolutionVisitor1926 auto retval = GetSpatialDimsToSplit(consumer->mutable_operand(0)); in Propagate()2009 auto retval = GetSpatialDimsToSplit(consumer->mutable_operand(0)); in Propagate()2545 auto old_new_dims = GetSpatialDimsToSplit(activations_old); in PropagateOnConv()2969 auto retval = GetSpatialDimsToSplit(producer); in PropagateOnConstant()