Home
last modified time | relevance | path

Searched refs:parameter_number_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc55 parameter_number_++, parameter->shape(), parameter->name()); in HandleParameter()
66 parameter_number_, hlo->shape(), hlo->name()); in DefaultAction()
67 parameter_number_++; in DefaultAction()
109 int64_t parameter_number_ = 0; member in xla::__anon1b44898a0111::ExtractionVisitor
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.cc89 if (instruction->parameter_number() != parameter_number_) { in MatchAndExplain()
92 << parameter_number_ << ")"; in MatchAndExplain()
Dbuffer_assignment.h127 return parameter_number_; in parameter_number()
281 parameter_number_ = parameter_number; in set_entry_computation_parameter()
323 int64_t parameter_number_ = 0; variable
Dhlo_matchers.h51 parameter_number_(parameter_number) {}
57 int64_t parameter_number_;
Dhlo_instructions.cc2203 parameter_number_(parameter_number) { in HloParameterInstruction()
2209 proto.set_parameter_number(parameter_number_); in ToProto()
2241 return StrCat(parameter_number_); in OperandsToStringWithCanonicalNameMap()
2256 auto clone = std::make_unique<HloParameterInstruction>(parameter_number_, in CloneWithNewOperandsImpl()
Dhlo_instructions.h1392 int64_t parameter_number() const { return parameter_number_; } in parameter_number()
1438 int64_t parameter_number_ = 0; variable
Dbuffer_assignment.cc267 proto.set_parameter_number(parameter_number_); in ToProto()