Home
last modified time | relevance | path

Searched defs:lhs_batch_dimensions (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc610 absl::Span<const int64_t> lhs_batch_dimensions = in ValidateDotDimensionNumbers() local
706 const auto& lhs_batch_dimensions = dimension_numbers.lhs_batch_dimensions(); in InferDotOpShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto619 repeated int64 lhs_batch_dimensions = 3; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc360 auto lhs_batch_dimensions = in Convert_dot_dimension_numbers() local