Searched refs:parameter_number_ (Results 1 – 7 of 7) sorted by relevance
53 parameter_number_++, parameter->shape(), parameter->name()); in HandleParameter()64 parameter_number_, hlo->shape(), hlo->name()); in DefaultAction()65 parameter_number_++; in DefaultAction()107 int64 parameter_number_ = 0; member in xla::__anonc6eaaf4e0111::ExtractionVisitor
83 if (instruction->parameter_number() != parameter_number_) { in MatchAndExplain()86 << parameter_number_ << ")"; in MatchAndExplain()
126 return parameter_number_; in parameter_number()285 parameter_number_ = parameter_number; in set_entry_computation_parameter()319 int64 parameter_number_ = 0; variable
48 parameter_number_(parameter_number) {}54 int64 parameter_number_;
1642 parameter_number_(parameter_number) { in HloParameterInstruction()1648 proto.set_parameter_number(parameter_number_); in ToProto()1676 return StrCat(parameter_number_); in OperandsToStringWithCanonicalNameMap()1691 return absl::make_unique<HloParameterInstruction>(parameter_number_, shape, in CloneWithNewOperandsImpl()
866 int64 parameter_number() const { return parameter_number_; } in parameter_number()901 int64 parameter_number_ = 0; variable
265 proto.set_parameter_number(parameter_number_); in ToProto()