OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:fused_root
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ops.h
43
HloInstruction*
fused_root
= fusion->fused_expression_root();
local
44
if (
fused_root
->opcode() != HloOpcode::kDynamicUpdateSlice ||
53
fused_root
->mutable_operand(0)->LatestNonGteAncestorAndIndex();
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
instruction_fusion_test.cc
159
HloInstruction*
fused_root
= root->fused_expression_root();
local
160
EXPECT_EQ(HloOpcode::kAdd,
fused_root
->opcode());
161
// Check that operands of '
fused_root
' are GTE.
162
EXPECT_EQ(HloOpcode::kGetTupleElement,
fused_root
->operand(0)->opcode());
163
EXPECT_EQ(HloOpcode::kGetTupleElement,
fused_root
->operand(1)->opcode());
/external/tensorflow/tensorflow/compiler/xla/service/
hlo_verifier.cc
632
const HloInstruction*
fused_root
= fusion->fused_expression_root();
local
635
if (
fused_root
== instruction) {
665
if (
fused_root
->user_count() != 0) {
674
if (instruction !=
fused_root
) {
[
all
...]
hlo_instruction.cc
779
const Shape& shape, FusionKind fusion_kind, HloInstruction*
fused_root
) {
783
instruction->set_parent(
fused_root
->parent());
784
instruction->set_metadata(
fused_root
->metadata());
785
instruction->CloneAndFuseInternal(
fused_root
);
1021
HloInstruction*
fused_root
= fused_expression_root();
local
[
all
...]
Completed in 4279 milliseconds