HomeSort by relevance Sort by last modified time
    Searched refs:param_instructions_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_computation.h 128 int64 num_parameters() const { return param_instructions_.size(); }
133 CHECK_LT(param_no, static_cast<int64>(param_instructions_.size()))
135 return param_instructions_[param_no];
139 return param_instructions_;
393 std::vector<HloInstruction*> param_instructions_; member in class:xla::HloComputation
hlo_computation.cc 70 param_instructions_.resize(parameter_count, nullptr);
78 CHECK(param_instructions_[param_no] == nullptr)
81 param_instructions_[param_no] = instruction.get();
115 CHECK(fusion_instruction_->operand_count() == param_instructions_.size());
117 param_instructions_.push_back(instruction.get());
124 CHECK_LT(param_no, param_instructions_.size());
126 HloInstruction* param_instruction = param_instructions_[param_no];
127 auto param_instruction_iterator = param_instructions_.begin() + param_no;
128 param_instructions_.erase(param_instruction_iterator);
132 while (param_no < param_instructions_.size())
    [all...]

Completed in 139 milliseconds