Home
last modified time | relevance | path

Searched defs:output_spatial_dims (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_rewriter.cc110 auto output_spatial_dims = conv_dnums.output_spatial_dimensions(); in MatchBackwardFilter() local
347 const auto& output_spatial_dims = dnums.output_spatial_dimensions(); in MatchBackwardInput() local
H A Djitrt_custom_calls.h56 llvm::ArrayRef<int64_t> output_spatial_dims; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dattribute_importer.cc96 llvm::SmallVector<int64_t, 4> output_spatial_dims( in ConvertConvDimensionNumbers() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc2026 std::vector<int64_t> output_spatial_dims = in ConvertConv2DHelper() local