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