Searched defs:collapsed_slice_dims (Results 1 – 9 of 9) sorted by relevance
52 std::vector<int64_t> collapsed_slice_dims( in ConvertGatherDimensionNumbers() local
267 auto collapsed_slice_dims = CreateDenseIntElementsAttrFromVector( in CreateGatherDimensionNumbers() local
302 auto collapsed_slice_dims = ConvertDenseIntAttr(input.collapsed_slice_dims()); in Convert_dimension_numbers() local
950 std::vector<int64> collapsed_slice_dims(dnums.collapsed_slice_dims().begin(), in GatherOutputShardingFromDataOperand() local964 std::vector<int64> collapsed_slice_dims(dnums.collapsed_slice_dims().begin(), in GatherDataOperandShardingFromOutput() local
2741 string collapsed_slice_dims = StrCat( in GatherDimensionNumbersToString() local2757 absl::Span<const int64> collapsed_slice_dims, in MakeGatherDimNumbers()
2444 optional<std::vector<int64>> collapsed_slice_dims; in ParseInstructionRhs() local
488 repeated int64 collapsed_slice_dims = 2; field
1039 auto collapsed_slice_dims = in matchAndRewrite() local
241 auto collapsed_slice_dims = llvm::to_vector<8>(llvm::map_range( in matchAndRewrite() local