Searched refs:WhileContext (Results 1 – 14 of 14) 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 | 64 class WhileContext; variable
206 WhileContext* while_ctx() const { return while_ctx_; } in while_ctx()
207 void set_while_ctx(WhileContext* while_ctx) { in set_while_ctx()
300 WhileContext* while_ctx_;
649 WhileContext** result);
724 std::map<string, WhileContext> while_ctxs_;
|
| D | graph.cc | 798 WhileContext** result) { in AddWhileContext()
799 auto pair = while_ctxs_.insert(std::pair<string, WhileContext>( in AddWhileContext()
801 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 | 721 self._grad_context = WhileContext(
735 self._grad_context = WhileContext(
2162 class WhileContext(ControlFlowContext): class
2271 super(WhileContext, self).__init__(
2357 super(WhileContext, self)._to_values_def(export_scope=export_scope))
2380 ret = WhileContext(context_def=context_def, import_scope=import_scope)
2693 if (isinstance(self.outer_context, WhileContext) and
3467 loop_context = WhileContext(
4053 return WhileContext.from_proto(
4068 to_proto=WhileContext.to_proto,
[all …]
|
| D | control_flow_ops_test.py | 450 control_flow_ops.WhileContext.from_proto(
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | control_flow.proto | 51 // 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 | 111 assert isinstance(self._while_context, control_flow_ops.WhileContext)
|