OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:HloComputation
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
hlo_computation.h
52
// An
HloComputation
contains a directed acyclic graph of HLO instructions. The
55
class
HloComputation
{
57
// Builder class for
HloComputation
.
66
// Build and return an
HloComputation
. The parameter root_instruction
70
std::unique_ptr<
HloComputation
> Build(
163
// computation_map: a map from computation name to
HloComputation
*. This map
171
static StatusOr<std::unique_ptr<
HloComputation
>> CreateFromProto(
173
const tensorflow::gtl::FlatMap<string,
HloComputation
*>& computation_map,
174
const std::function<void(std::unique_ptr<
HloComputation
>)>&
222
std::list<
HloComputation
*> MakeEmbeddedComputationsList() const
[
all
...]
hlo_computation.cc
47
std::unique_ptr<
HloComputation
>
HloComputation
::Builder::Build(
59
return WrapUnique(new
HloComputation
(name_, parameter_count, &instructions_,
63
HloComputation
::
HloComputation
(
90
HloInstruction*
HloComputation
::AddInstruction(
98
HloInstruction*
HloComputation
::AddInstructionInternal(
111
HloInstruction*
HloComputation
::AddParameter(
122
Status
HloComputation
::RemoveParameter(int64 param_no) {
161
void
HloComputation
::Reparent(HloInstruction* instruction)
[
all
...]
Completed in 90 milliseconds