Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc753 std::vector<int64_t> lhs_batch_dimensions( in ConvertDotDimensionNumbers() local
Dmlir_hlo_to_hlo.cc251 auto lhs_batch_dimensions = in Convert_dot_dimension_numbers() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto543 repeated int64 lhs_batch_dimensions = 3; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc592 absl::Span<const int64> lhs_batch_dimensions = in ValidateDotDimensionNumbers() local