Searched refs:WhileContext (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 57 Status AddForwardLoopCounter(WhileContext* while_ctx, const Scope& scope, in AddForwardLoopCounter()
96 Status AddBackPropLoopCounter(WhileContext* while_ctx, const Output& loop_count, in AddBackPropLoopCounter()
136 Status AddWhileGradientLoop(WhileContext* while_ctx, in AddWhileGradientLoop()
179 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope, in AddWhileLoopGradient()
|
| D | while_gradients.h | 34 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope,
|
| D | gradients.cc | 135 std::map<WhileContext*, std::map<Node*, Output>> while_backprops_;
402 WhileContext* while_ctx = exit_node->while_ctx(); in ProcessWhileLoop()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | while_context.cc | 20 WhileContext::WhileContext(StringPiece frame_name, in WhileContext() function in tensorflow::WhileContext
|
| D | while_context.h | 35 class WhileContext {
37 WhileContext(StringPiece frame_name, std::vector<Node*> enter_nodes,
|
| D | graph.h | 63 class WhileContext; variable
186 WhileContext* while_ctx() const { return while_ctx_; } in while_ctx()
187 void set_while_ctx(WhileContext* while_ctx) { in set_while_ctx()
270 WhileContext* while_ctx_;
584 WhileContext** result);
656 std::map<string, WhileContext> while_ctxs_;
|
| D | graph.cc | 690 WhileContext** result) { in AddWhileContext()
691 auto pair = while_ctxs_.insert(std::pair<string, WhileContext>( in AddWhileContext()
693 WhileContext(frame_name, std::move(enter_nodes), std::move(exit_nodes), in AddWhileContext()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_grad.py | 47 if isinstance(op_ctxt, WhileContext):
99 if isinstance(op_ctxt, WhileContext):
|
| D | control_flow_ops.py | 854 self._grad_context = WhileContext(
868 self._grad_context = WhileContext(
2192 class WhileContext(ControlFlowContext): class
2300 super(WhileContext, self).__init__(
2385 super(WhileContext, self)._to_values_def(
2413 ret = WhileContext(context_def=context_def,
2711 if (isinstance(self.outer_context, WhileContext) and
3260 loop_context = WhileContext(
3733 return WhileContext.from_proto(context_def.while_ctxt,
3748 to_proto=WhileContext.to_proto,
[all …]
|
| D | control_flow_ops_test.py | 508 control_flow_ops.WhileContext.from_proto(
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | control_flow.proto | 38 // Protocol buffer representing a WhileContext object.
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop_test.cc | 97 WhileContext* while_ctx; in TEST_F()
|
| D | while_loop.cc | 234 WhileContext* while_ctx; in BuildWhileLoop()
|