Home
last modified time | relevance | path

Searched defs:output_feature_dim (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dconv_canonicalization.cc106 auto output_feature_dim = dnums.output_feature_dimension(); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.h55 int64_t output_feature_dim; member
H A Dcudnn_simplify_padding.cc273 int64_t output_feature_dim; in TrySimplifyPadding() local
H A Dgpu_conv_rewriter.cc109 auto output_feature_dim = conv_dnums.output_feature_dimension(); in MatchBackwardFilter() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dconv_op_helpers.cc51 int64_t output_feature_dim = filter_shape.dimensions_size() - 1; in GroupedFilterShapeForDepthwiseConvolution() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_cost_analysis.cc599 const int64_t output_feature_dim = dnums.output_feature_dimension(); in GetConvolutionFlops() local