Home
last modified time | relevance | path

Searched defs:branch_computation (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc942 for (auto branch_computation : instruction->branch_computations()) { in CheckLayouts() local
Dhlo_instruction.cc2080 HloComputation* HloInstruction::branch_computation(int b) const { in branch_computation() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2089 for (auto branch_computation : conditional->branch_computations()) { in CheckConditionalBuffersShareAllocation() local
2162 HloComputation* branch_computation = hlo->branch_computation(j); in BuildConditionalThunk() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1925 for (const XlaComputation* branch_computation : branch_computations) { in Conditional() local