Home
last modified time | relevance | path

Searched defs:kernel_vect_dim (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_vectorize_convolutions.cc228 int64_t kernel_vect_dim = dnums.kernel_input_feature_dimension(); in VectorizeDnums() local
270 std::optional<int64_t> kernel_vect_dim; in TryRevectorizeConv() local
H A Dcudnn_pad_for_convolutions.cc326 std::optional<int64_t> kernel_vect_dim; in TryResolvePaddedShapesForIntegerConvolution() local