Searched defs:start_index_map (Results 1 – 10 of 10) sorted by relevance
69 std::vector<int64_t> start_index_map(dims.start_index_map().size()); in ExpandInstruction() local
1068 std::vector<int64_t> start_index_map(dnums.start_index_map().begin(), in GatherOutputShardingFromDataOperand() local1082 std::vector<int64_t> start_index_map(dnums.start_index_map().begin(), in GatherDataOperandShardingFromOutput() local
3068 std::string start_index_map = in GatherDimensionNumbersToString() local3082 absl::Span<const int64_t> start_index_map, int64_t index_vector_dim) { in MakeGatherDimNumbers()
2810 optional<std::vector<int64_t>> start_index_map; in CreateInstruction() local
52 std::vector<int64_t> start_index_map(dnums.start_index_map().begin(), in ConvertGatherDimensionNumbers() local
423 auto start_index_map = input.getStartIndexMap(); in Convert_dimension_numbers() local
541 repeated int64 start_index_map = 3; field
381 std::vector<int64_t> start_index_map(dnums.start_index_map().begin(), in PartitionGatherTrivialIndexedOperandDimension() local
2904 auto start_index_map = gather_op.dimension_numbers().getStartIndexMap(); in matchAndRewrite() local
1547 SmallVector<int64_t, 4> start_index_map; in matchAndRewrite() local2044 SmallVector<int64_t, 2> start_index_map({num_dims - 2, num_dims - 1}); in matchAndRewrite() local