Searched defs:offset_dims (Results 1 – 10 of 10) sorted by relevance
50 std::vector<int64_t> offset_dims(dnums.offset_dims().begin(), in ConvertGatherDimensionNumbers() local
263 auto offset_dims = CreateDenseIntElementsAttrFromVector( in CreateGatherDimensionNumbers() local
297 auto offset_dims = ConvertDenseIntAttr(input.offset_dims()); in Convert_dimension_numbers() local
253 HloInstruction* accumulator, absl::Span<const int64> offset_dims, in PermuteBatchAndOffsetDims()
954 std::vector<int64> offset_dims(dnums.offset_dims().begin(), in GatherOutputShardingFromDataOperand() local968 std::vector<int64> offset_dims(dnums.offset_dims().begin(), in GatherDataOperandShardingFromOutput() local
2738 string offset_dims = in GatherDimensionNumbersToString() local2756 absl::Span<const int64> offset_dims, in MakeGatherDimNumbers()
2441 optional<std::vector<int64>> offset_dims; in ParseInstructionRhs() local
487 repeated int64 offset_dims = 1; field
1070 auto offset_dims = gather_op.dimension_numbers().offset_dims(); in matchAndRewrite() local
3399 std::vector<int64> offset_dims = {x_dims[is_nhwc ? 3 : 1]}; in TEST_F() local