Searched refs:WhileContext (Results 1 – 15 of 15) 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 | 136 std::map<WhileContext*, std::map<Node*, Output>> while_backprops_;
403 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 | 68 class WhileContext; variable
234 WhileContext* while_ctx() const { return while_ctx_; } in while_ctx()
235 void set_while_ctx(WhileContext* while_ctx) { in set_while_ctx()
347 WhileContext* while_ctx_;
700 WhileContext** result);
793 std::map<string, WhileContext> while_ctxs_;
|
| D | graph.cc | 855 WhileContext** result) { in AddWhileContext()
856 auto pair = while_ctxs_.insert(std::pair<string, WhileContext>( in AddWhileContext()
858 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):
104 if isinstance(op_ctxt, WhileContext):
|
| D | control_flow_ops.py | 1453 class WhileContext(ControlFlowContext): class
1562 super(WhileContext, self).__init__(
1648 super(WhileContext, self)._to_values_def(export_scope=export_scope))
1671 ret = WhileContext(context_def=context_def, import_scope=import_scope)
1993 if (isinstance(self.outer_context, WhileContext) and
2806 loop_context = WhileContext(
3765 return WhileContext.from_proto(
3780 to_proto=WhileContext.to_proto,
3781 from_proto=WhileContext.from_proto)
|
| D | control_flow_state.py | 182 self._grad_context = control_flow_ops.WhileContext(
196 self._grad_context = control_flow_ops.WhileContext(
|
| D | control_flow_ops_test.py | 496 control_flow_ops.WhileContext.from_proto(
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | control_flow.proto | 52 // 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()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 220 assert isinstance(self._while_context, control_flow_ops.WhileContext)
|