Home
last modified time | relevance | path

Searched defs:input_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.cc218 int64_t input_vect_dim = dnums.input_feature_dimension(); in VectorizeDnums() local
269 std::optional<int64_t> input_vect_dim; in TryRevectorizeConv() local
H A Dcudnn_pad_for_convolutions.cc325 std::optional<int64_t> input_vect_dim; in TryResolvePaddedShapesForIntegerConvolution() local