OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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