Searched defs:DotDimensionNumbers (Results 1 – 4 of 4) sorted by relevance
38 struct DotDimensionNumbers { struct39 llvm::ArrayRef<int64_t> lhs_batch;40 llvm::ArrayRef<int64_t> lhs_contract;41 llvm::ArrayRef<int64_t> rhs_batch;42 llvm::ArrayRef<int64_t> rhs_contract;
465 class DotDimensionNumbers: class478 dimension_numbers: Union[DotDimensionNumbers,
322 DotDimensionNumbers = xla_data_pb2.DotDimensionNumbers variable
613 message DotDimensionNumbers { message