Home
last modified time | relevance | path

Searched defs:rhs_dimension (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc322 int64_t rhs_dimension = dnums.kernel_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
359 int64_t rhs_dimension = dnums.kernel_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
631 int64_t rhs_dimension = dnums.kernel_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dtridiagonal.cc134 const int64_t rhs_dimension = ShapeUtil::GetDimension(rhs_shape, i); in ValidateTridiagonalMatMulDiagonal() local