Searched refs:convolution_dimension_numbers_ (Results 1 – 2 of 2) sorted by relevance
1863 convolution_dimension_numbers_(dimension_numbers), in HloConvolutionInstruction()1890 convolution_dimension_numbers_; in ToProto()1904 convolution_dimension_numbers_))); in ExtraAttributesToStringImpl()1948 batch_group_count_, window(), convolution_dimension_numbers_, in CloneWithNewOperandsImpl()2080 if (convolution_dimension_numbers_ != nullptr) { in ToProto()2082 *convolution_dimension_numbers_; in ToProto()2103 if (convolution_dimension_numbers_ != nullptr) { in ExtraAttributesToStringImpl()2106 ConvolutionDimensionNumbersToString(*convolution_dimension_numbers_))); in ExtraAttributesToStringImpl()2146 if ((convolution_dimension_numbers_ == nullptr) != in IdenticalSlowPath()2147 (casted_other.convolution_dimension_numbers_ == nullptr) || in IdenticalSlowPath()[all …]
1039 return convolution_dimension_numbers_; in convolution_dimension_numbers()1043 convolution_dimension_numbers_ = dnums; in set_convolution_dimension_numbers()1087 ConvolutionDimensionNumbers convolution_dimension_numbers_; variable1192 CHECK(convolution_dimension_numbers_ != nullptr); in convolution_dimension_numbers()1193 return *convolution_dimension_numbers_; in convolution_dimension_numbers()1198 convolution_dimension_numbers_ = in set_convolution_dimension_numbers()1242 std::unique_ptr<ConvolutionDimensionNumbers> convolution_dimension_numbers_; variable