Searched refs:CreateGetDimensionSize (Results 1 – 4 of 4) sorted by relevance
116 add_instruction(HloInstruction::CreateGetDimensionSize( in DynamicElementCountPerFeature()
1034 static std::unique_ptr<HloInstruction> CreateGetDimensionSize(
735 CreateGetDimensionSize(shape, operands(0), proto.dimensions(0)); in CreateFromProto()1369 HloInstruction::CreateGetDimensionSize(const Shape& shape, in CreateGetDimensionSize() function in xla::HloInstruction
2578 builder->AddInstruction(HloInstruction::CreateGetDimensionSize( in ParseInstructionRhs()