Searched refs:CreateSetDimensionSize (Results 1 – 6 of 6) sorted by relevance
1266 builder.AddInstruction(HloInstruction::CreateSetDimensionSize( in TEST_F()1293 input = builder.AddInstruction(HloInstruction::CreateSetDimensionSize( in TEST_F()1297 input = builder.AddInstruction(HloInstruction::CreateSetDimensionSize( in TEST_F()
552 reshape->AddInstruction(HloInstruction::CreateSetDimensionSize( in RewriteDynamicReshapeSplitInput()578 reshape->AddInstruction(HloInstruction::CreateSetDimensionSize( in RewriteDynamicReshapeSplitInput()736 reshape->AddInstruction(HloInstruction::CreateSetDimensionSize( in RewriteDynamicReshapeCombineInput()749 gather = reshape->AddInstruction(HloInstruction::CreateSetDimensionSize( in RewriteDynamicReshapeCombineInput()
1168 static std::unique_ptr<HloInstruction> CreateSetDimensionSize(
918 instruction = CreateSetDimensionSize(shape, operands(0), operands(1), in CreateFromProto()1676 HloInstruction::CreateSetDimensionSize(const Shape& shape, in CreateSetDimensionSize() function in xla::HloInstruction
1219 HloInstruction::CreateSetDimensionSize(vector_shape_, param, size, 0)); in TEST_P()
2958 return builder->AddInstruction(HloInstruction::CreateSetDimensionSize( in CreateInstruction()