Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto584 repeated int64 kernel_spatial_dimensions = 6; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc882 auto kernel_spatial_dimensions = dnums.getKernelSpatialDimensions(); in IsSupportedConvOp() local
917 auto kernel_spatial_dimensions = dnums.getKernelSpatialDimensions(); in MatchResizeOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc274 std::vector<int64_t> kernel_spatial_dimensions( in GenNewConvDNums() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc997 SmallVector<int64_t, 4> kernel_spatial_dimensions; in GetConvDimensionNumbersAttr() local