Home
last modified time | relevance | path

Searched defs:kernel_output_feature_dim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc43 auto kernel_output_feature_dim = dnums.kernel_output_feature_dimension(); in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc165 int64 kernel_output_feature_dim, int64 group_count, in GetExpandedFilterMask()
338 int64 kernel_output_feature_dim = in HandleConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter.cc122 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension(); in MatchBackwardFilter() local