Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto621 repeated int64 rhs_batch_dimensions = 4; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc612 absl::Span<const int64_t> rhs_batch_dimensions = in ValidateDotDimensionNumbers() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc358 auto rhs_batch_dimensions = in Convert_dot_dimension_numbers() local