Home
last modified time | relevance | path

Searched refs:DimNums (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.h35 struct DimNums { struct
43 std::vector<DimNums> batch_dims;
44 std::vector<DimNums> contracting_dims;
45 std::vector<DimNums> lhs_non_contracting_dims;
46 std::vector<DimNums> rhs_non_contracting_dims;
47 std::vector<DimNums> conv_spatial_dims;
Dsharding_propagation.cc483 const dot_as_convolution_util::DotConvolutionDimsInfo::DimNums> in InferConvolutionShardingFromOperands()