Searched refs:positions_ (Results 1 – 3 of 3) sorted by relevance
75 positions_.push_back(HloPosition{instruction, index}); in HloValue()154 CHECK_EQ(positions_.size(), 1) << "SetPositions should only be called once."; in SetPositionsAndComputeUses()167 positions_.insert(positions_.end(), positions.begin(), positions.end()); in SetPositionsAndComputeUses()171 for (const HloPosition& position : positions_) { in SetPositionsAndComputeUses()179 for (const HloPosition& position : positions_) { in SetPositionsAndComputeUses()
129 const HloPosition& defining_position() const { return positions_[0]; } in defining_position()150 const std::vector<HloPosition>& positions() const { return positions_; } in positions()174 std::vector<HloPosition> positions_; variable
38 positions_ = AddInput(positions); in GatherOpModel()42 BuildInterpreter({GetShape(input_), GetShape(positions_)}); in GatherOpModel()56 PopulateTensor<T>(positions_, data); in SetPositions()71 int positions_; member in tflite::__anonfb8da32a0111::GatherOpModel