OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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