HomeSort by relevance Sort by last modified time
    Searched refs:WhileContext (Results 1 - 12 of 12) sorted by null

  /external/tensorflow/tensorflow/cc/framework/
while_gradients.h 34 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope,
while_gradients.cc 57 Status AddForwardLoopCounter(WhileContext* while_ctx, const Scope& scope,
96 Status AddBackPropLoopCounter(WhileContext* while_ctx, const Output& loop_count,
136 Status AddWhileGradientLoop(WhileContext* while_ctx,
179 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope,
gradients.cc 135 std::map<WhileContext*, std::map<Node*, Output>> while_backprops_;
402 WhileContext* while_ctx = exit_node->while_ctx();
  /external/tensorflow/tensorflow/core/graph/
while_context.cc 20 WhileContext::WhileContext(StringPiece frame_name,
while_context.h 24 // associated WhileContext, i.e., there is a WhileContext for every execution
35 class WhileContext {
37 WhileContext(StringPiece frame_name, std::vector<Node*> enter_nodes,
graph.h 63 class WhileContext;
186 WhileContext* while_ctx() const { return while_ctx_; }
187 void set_while_ctx(WhileContext* while_ctx) {
266 // WhileContext. Otherwise null. (This is only set for exit nodes because
270 WhileContext* while_ctx_;
576 // Create and return a new WhileContext owned by this graph. This is called
584 WhileContext** result);
656 std::map<string, WhileContext> while_ctxs_;
graph.cc 690 WhileContext** result) {
691 auto pair = while_ctxs_.insert(std::pair<string, WhileContext>(
693 WhileContext(frame_name, std::move(enter_nodes), std::move(exit_nodes),
698 return errors::InvalidArgument("WhileContext with frame name '", frame_name,
  /external/tensorflow/tensorflow/python/ops/
control_flow_grad.py 47 if isinstance(op_ctxt, WhileContext):
99 if isinstance(op_ctxt, WhileContext):
control_flow_ops.py 776 # Find the next outer WhileContext (or stop if we reach the
    [all...]
control_flow_ops_test.py 508 control_flow_ops.WhileContext.from_proto(
    [all...]
  /external/tensorflow/tensorflow/cc/ops/
while_loop.cc 234 WhileContext* while_ctx;
while_loop_test.cc 97 WhileContext* while_ctx;

Completed in 2266 milliseconds