Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.h443 struct DotConvDimsMapping { struct
448 struct DimsMapping {
455 std::vector<DimsMapping> batch_dims;
456 std::vector<DimsMapping> contracting_dims;
457 std::vector<DimsMapping> lhs_non_contracting_dims;
458 std::vector<DimsMapping> rhs_non_contracting_dims;
459 std::vector<DimsMapping> conv_spatial_dims;