Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:fused_expression_root (Results 1 - 24 of 24) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/
liveness_util.cc	163 fused_param_uses[0].first == fusion->fused_expression_root() && 199 user->fused_expression_root()->opcode() == 209 user->fused_expression_root()->opcode() == HloOpcode::kAdd) { 214 auto* add = user->fused_expression_root(); 304 user->fused_expression_root()->opcode() == 311 return use.instruction == user->fused_expression_root() && 314 user->fused_expression_root()->opcode() == HloOpcode::kAdd) { 319 auto* add = user->fused_expression_root(); 339 return use.instruction == user->fused_expression_root() &&
defuser.cc	81 defused_instructions.at(fusion_instruction->fused_expression_root())));
reduce_precision_insertion_test.cc	390 HloInstruction* y_fused = z->fused_expression_root(); 404 EXPECT_EQ(z->fused_expression_root(), y_fused); 426 HloInstruction* y_fused = z->fused_expression_root(); 450 EXPECT_THAT(z->fused_expression_root(), y_fused); 473 HloInstruction* y_fused = z->fused_expression_root(); 497 EXPECT_THAT(z->fused_expression_root(), op::ReducePrecision(y_fused));
logical_buffer_analysis.cc	72 TF_RETURN_IF_ERROR(instruction->fused_expression_root()->Accept(this));
reduce_precision_insertion.cc	180 HloInstruction* fusion_root = instruction->fused_expression_root();
defuser_test.cc	199 HloInstruction* fused_negate = outer_fusion->fused_expression_root();
hlo_instruction_test.cc	700 metadata, fusion->fused_expression_root()->metadata())); 702 metadata, fusion->fused_expression_root()->operand(0)->metadata()));     [all...]
hlo_instruction.h	[all...]
layout_assignment_test.cc	174 layout, fusion->fused_expression_root()->shape().layout())); 178 fusion->fused_expression_root()->operand(0)->shape()));     [all...]
tuple_points_to_analysis.cc	170 TF_RETURN_IF_ERROR(instruction->fused_expression_root()->Accept(this));
hlo_instruction.cc	2473 HloInstruction* HloInstruction::fused_expression_root() const { function in class:HloInstruction     [all...]
layout_assignment.cc	742 fusion->shape(), fusion->fused_expression_root()->shape()));     [all...]
hlo_verifier.cc	632 const HloInstruction* fused_root = fusion->fused_expression_root();     [all...]
hlo_graph_dumper.cc	[all...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
fusion_merger.cc	102 for (auto& operand : fusion->fused_expression_root()->operands()) { 106 bytes += ShapeUtil::ByteSizeOf(fusion->fused_expression_root()->shape()); 111 TF_CHECK_OK(fusion->fused_expression_root()->Accept(&analysis)); 123 for (auto& operand : fusion->fused_expression_root()->operands()) { 128 ShapeUtil::ByteSizeOf(fusion->fused_expression_root()->shape());
instruction_fusion_test.cc	159 HloInstruction* fused_root = root->fused_expression_root();
ir_emission_utils.cc	86 hlo.fused_expression_root()->opcode() == HloOpcode::kDot) {
while_transformer.cc	139 instruction->fused_expression_root(), tagged_instructions));
ir_emitter_unnested.cc	493 HloInstruction* root = fusion->fused_expression_root(); 560 << fusion->fused_expression_root()->opcode();     [all...]
ir_emitter.cc	695 TF_RETURN_IF_ERROR(fusion->fused_expression_root()->Accept(&fused_emitter));     [all...]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ops.h	43 HloInstruction* fused_root = fusion->fused_expression_root();
ops.cc	125 auto* dynamic_update_slice = fusion->fused_expression_root();
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc	[all...]
ir_emitter.cc	[all...]

Completed in 194 milliseconds