Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto594 repeated int64 output_spatial_dimensions = 12; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc870 auto output_spatial_dimensions = dnums.getOutputSpatialDimensions(); in IsSupportedConvOp() local
918 auto output_spatial_dimensions = dnums.getOutputSpatialDimensions(); in MatchResizeOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc298 std::vector<int64_t> output_spatial_dimensions( in GenNewConvDNums() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc5242 SmallVector<int64_t, 4> output_spatial_dimensions; in matchAndRewrite() local