Searched defs:collapsed_slice_dims (Results 1 – 11 of 11) sorted by relevance
52 std::vector<int64_t> collapsed_slice_dims( in ConvertGatherDimensionNumbers() local
274 auto collapsed_slice_dims = CreateDenseIntElementsAttrFromVector( in CreateGatherDimensionNumbers() local
326 auto collapsed_slice_dims = ConvertDenseIntAttr(input.collapsed_slice_dims()); in Convert_dimension_numbers() local
998 std::vector<int64> collapsed_slice_dims(dnums.collapsed_slice_dims().begin(), in GatherOutputShardingFromDataOperand() local1012 std::vector<int64> collapsed_slice_dims(dnums.collapsed_slice_dims().begin(), in GatherDataOperandShardingFromOutput() local
2905 string collapsed_slice_dims = StrCat( in GatherDimensionNumbersToString() local2921 absl::Span<const int64> collapsed_slice_dims, in MakeGatherDimNumbers()
2600 optional<std::vector<int64>> collapsed_slice_dims; in ParseInstructionRhs() local
271 SmallVector<int64_t, 4> collapsed_slice_dims; in matchAndRewrite() local
494 repeated int64 collapsed_slice_dims = 2; field
1644 auto collapsed_slice_dims = in matchAndRewrite() local
317 auto collapsed_slice_dims = llvm::to_vector<8>(llvm::map_range( in matchAndRewrite() local401 SmallVector<int64_t, 4> collapsed_slice_dims( in GatherShapeInferImpl() local
1206 SmallVector<int64_t, 4> collapsed_slice_dims(1, axis); in matchAndRewrite() local1711 SmallVector<int64_t, 4> collapsed_slice_dims; in matchAndRewrite() local