Searched refs:HloConvolutionInstruction (Results 1 – 4 of 4) sorted by relevance
2133 HloConvolutionInstruction::HloConvolutionInstruction( in HloConvolutionInstruction() function in xla::HloConvolutionInstruction2154 string HloConvolutionInstruction::ToCategory() const { in ToCategory()2165 HloInstructionProto HloConvolutionInstruction::ToProto() const { in ToProto()2176 std::vector<string> HloConvolutionInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()2199 bool HloConvolutionInstruction::IdenticalSlowPath( in IdenticalSlowPath()2204 static_cast<const HloConvolutionInstruction&>(other); in IdenticalSlowPath()2220 HloConvolutionInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()2224 return absl::make_unique<HloConvolutionInstruction>( in CloneWithNewOperandsImpl()
988 return absl::make_unique<HloConvolutionInstruction>( in CreateConvolve()3853 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config()3867 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config()4146 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in convolution_dimension_numbers()4157 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_convolution_dimension_numbers()4167 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in feature_group_count()4174 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_feature_group_count()4182 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in batch_group_count()4189 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_batch_group_count()
1224 class HloConvolutionInstruction : public HloInstruction {1226 explicit HloConvolutionInstruction(
4029 EXPECT_THAT(Cast<HloConvolutionInstruction>(conv) in TEST_P()