Searched defs:collapsed_slice_dims (Results 1 – 9 of 9) sorted by relevance
88 std::vector<int64_t> collapsed_slice_dims( in ExpandInstruction() local
1066 std::vector<int64_t> collapsed_slice_dims( in GatherOutputShardingFromDataOperand() local1080 std::vector<int64_t> collapsed_slice_dims( in GatherDataOperandShardingFromOutput() local
3065 std::string collapsed_slice_dims = StrCat( in GatherDimensionNumbersToString() local3081 absl::Span<const int64_t> collapsed_slice_dims, in MakeGatherDimNumbers()
2807 optional<std::vector<int64_t>> collapsed_slice_dims; in CreateInstruction() local
50 std::vector<int64_t> collapsed_slice_dims( in ConvertGatherDimensionNumbers() local
418 auto collapsed_slice_dims = input.getCollapsedSliceDims(); in Convert_dimension_numbers() local
536 repeated int64 collapsed_slice_dims = 2; field
2905 auto collapsed_slice_dims = in matchAndRewrite() local
1535 SmallVector<int64_t, 4> collapsed_slice_dims; in matchAndRewrite() local