Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc88 std::vector<int64_t> rhs_batch_dimensions( in ConvertDotDimensionNumbers() local
Dmlir_hlo_to_hlo.cc264 auto rhs_batch_dimensions = in Convert_dot_dimension_numbers() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto579 repeated int64 rhs_batch_dimensions = 4; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc604 absl::Span<const int64> rhs_batch_dimensions = in ValidateDotDimensionNumbers() local