Home
last modified time | relevance | path

Searched refs:is_loop_invariant (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc136 if (arg.is_loop_invariant) { in BuildLoopCondition()
179 if (arg.is_loop_invariant) { in BuildLoopBody()
232 if (arg.is_loop_invariant) { in FunctionalizeLoop()
243 new_arg.is_loop_invariant = false; in FunctionalizeLoop()
292 if (!arg.is_loop_invariant) { in FunctionalizeLoop()
472 if (!arg.is_loop_invariant) { in FunctionalizeLoop()
Dfunctionalize_control_flow_util.h38 bool is_loop_invariant; member
Dconst_analysis.cc117 bool is_loop_invariant, in GetCompileTimeConstInputs()
120 if (is_loop_invariant) { in GetCompileTimeConstInputs()
Dfunctionalize_control_flow_util.cc70 &arg.is_loop_invariant)); in ExtractWhileLoopFrames()
Dtf2xla_util.cc360 TF_ASSIGN_OR_RETURN(bool is_loop_invariant, in TraverseUnmodifiedPathBackward()
363 if (!is_loop_invariant) { in TraverseUnmodifiedPathBackward()
444 bool is_loop_invariant, in PropagateConstIntoAndAroundWhileNode()
446 if (!is_loop_invariant) { in PropagateConstIntoAndAroundWhileNode()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc181 bool is_loop_invariant = node->IsSwitch() && in PropagateShapes() local
184 if (is_loop_invariant) { in PropagateShapes()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc126 StatusOr<bool> is_loop_invariant = IsLoopInvariant(body, i, fld); in GetLoopInvariants() local
127 OP_REQUIRES_OK(ctx, is_loop_invariant.status()); in GetLoopInvariants()
128 (*loop_invariants)[i] = *is_loop_invariant; in GetLoopInvariants()