Searched refs:convolution_dimension_numbers_ (Results 1 – 2 of 2) sorted by relevance
2142 convolution_dimension_numbers_(dimension_numbers), in HloConvolutionInstruction()2169 convolution_dimension_numbers_; in ToProto()2183 convolution_dimension_numbers_))); in ExtraAttributesToStringImpl()2226 batch_group_count_, window(), convolution_dimension_numbers_, in CloneWithNewOperandsImpl()2416 if (convolution_dimension_numbers_ != nullptr) { in ToProto()2418 *convolution_dimension_numbers_; in ToProto()2454 if (convolution_dimension_numbers_ != nullptr) { in ExtraAttributesToStringImpl()2457 ConvolutionDimensionNumbersToString(*convolution_dimension_numbers_))); in ExtraAttributesToStringImpl()2516 if ((convolution_dimension_numbers_ == nullptr) != in IdenticalSlowPath()2517 (casted_other.convolution_dimension_numbers_ == nullptr) || in IdenticalSlowPath()[all …]
1234 return convolution_dimension_numbers_; in convolution_dimension_numbers()1238 convolution_dimension_numbers_ = dnums; in set_convolution_dimension_numbers()1285 ConvolutionDimensionNumbers convolution_dimension_numbers_; variable1434 CHECK(convolution_dimension_numbers_ != nullptr); in convolution_dimension_numbers()1435 return *convolution_dimension_numbers_; in convolution_dimension_numbers()1440 convolution_dimension_numbers_ = in set_convolution_dimension_numbers()1522 std::unique_ptr<ConvolutionDimensionNumbers> convolution_dimension_numbers_; variable