Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.h38 struct DotDimensionNumbers { struct
39 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;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_client.py465 class DotDimensionNumbers: class
478 dimension_numbers: Union[DotDimensionNumbers,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/
H A Dxla.py322 DotDimensionNumbers = xla_data_pb2.DotDimensionNumbers variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto613 message DotDimensionNumbers { message