Home
last modified time | relevance | path

Searched defs:lhs_contracting_dimensions (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc85 std::vector<int64_t> lhs_contracting_dimensions( in ConvertDotDimensionNumbers() local
Dmlir_hlo_to_hlo.cc261 auto lhs_contracting_dimensions = in Convert_dot_dimension_numbers() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto573 repeated int64 lhs_contracting_dimensions = 1; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc598 absl::Span<const int64> lhs_contracting_dimensions = in ValidateDotDimensionNumbers() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc494 auto lhs_contracting_dimensions = GetI64ElementsAttr( in BatchDot() local
3461 auto lhs_contracting_dimensions = GetI64ElementsAttr( in matchAndRewrite() local