Searched defs:dot_dims (Results 1 – 8 of 8) sorted by relevance
205 const DotDimensionNumbers& dot_dims = dot.dot_dimension_numbers(); in CanFoldTransposeOperandIntoDot() local332 const DotDimensionNumbers& dot_dims = config.dot_dimension_numbers(); in For() local343 mlir::mhlo::DotDimensionNumbersAttr dot_dims = op.getDotDimensionNumbers(); in For() local613 mlir::mhlo::DotDimensionNumbersAttr dot_dims = op.getDotDimensionNumbers(); in For() local
55 const DotDimensionNumbers &dot_dims = config.dot_dimension_numbers(); in SetName() local312 const DotDimensionNumbers &dot_dims = config.dot_dimension_numbers(); in FuseVectorBiasAdd() local
232 const DotDimensionNumbers& dot_dims = dot->dot_dimension_numbers(); in DotCanSupportShapeWithLayout() local265 const DotDimensionNumbers& dot_dims = in AddBackendConstraints() local
521 DotDimensionNumbers dot_dims, int64_t uid) const { in operator ()()595 DotDimensionNumbers dot_dims, se::cuda::BlasLt::Epilogue epilogue, in operator ()()
404 xla::DotDimensionNumbers dot_dims; in CalculateGradWarp() local540 xla::DotDimensionNumbers dot_dims; in Compile() local
152 std::vector<int64_t> dot_dims = batch_dim_sizes; in CanonicalizeDot() local
261 const DotDimensionNumbers& dot_dims = dot.dot_dimension_numbers(); in IsRowColumnTransposeDotOperand() local
559 auto dot_dims = in InferConvolutionShardingFromOperands() local1379 auto dot_dims = dot_as_convolution_util::ParseConvolutionDimsInfo(&user); in GetShardingFromUser() local