experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make DotDimensionNumbers more specific.

Browse Source 
This makes client code easier to write.

PiperOrigin-RevId: 299451495
Change-Id: I840266f75ad2f8a6b987dfbb9aad9cff59ed5297
This commit is contained in:
Sean Silva 2020-03-06 15:39:22 -08:00 committed by TensorFlower Gardener
parent 326e067101
commit 34e58ac1bd
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/compiler/mlir/xla/ir/hlo_ops.td
View File

@ -944,10 +944,11 @@ def HLO_DotOp: HLO_Op<"dot", [NoSideEffect]>, BASE_HLO_DotOp {
}
def DotDimensionNumbers : StructAttr<"DotDimensionNumbers", HLO_Dialect, [
StructFieldAttr<"lhs_batching_dimensions", ElementsAttr>,
StructFieldAttr<"rhs_batching_dimensions", ElementsAttr>,
StructFieldAttr<"lhs_contracting_dimensions", ElementsAttr>,
StructFieldAttr<"rhs_contracting_dimensions", ElementsAttr>] > {
StructFieldAttr<"lhs_batching_dimensions", I64ElementsAttr>,
StructFieldAttr<"rhs_batching_dimensions", I64ElementsAttr>,
StructFieldAttr<"lhs_contracting_dimensions", I64ElementsAttr>,
StructFieldAttr<"rhs_contracting_dimensions", I64ElementsAttr>
]> {
let description = "Structure of dimension information for dot product";
}