Searched defs:body_fn (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 75 BodyGraphBuilderFn body_fn = [](const Scope& scope, in AddForwardLoopCounter() local
114 BodyGraphBuilderFn body_fn = [](const Scope& scope, in AddBackPropLoopCounter() local
160 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope, in AddWhileGradientLoop() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 354 NameAttrList cond_fn, body_fn; in TEST() local
388 NameAttrList cond_fn, body_fn; in TEST() local
|
| D | functionalize_control_flow_test.cc | 261 NameAttrList cond_fn, body_fn; in TEST() local
413 NameAttrList cond_fn, body_fn; in TEST() local
533 NameAttrList cond_fn, body_fn; in TEST() local
681 NameAttrList cond_fn, body_fn; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 68 auto body_fn = in CholeskyUnblocked() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 92 def body_fn(i): # pylint: disable=invalid-name function
|
| D | control_flow_ops_py_test.py | 1303 def body_fn(i): function
1320 def body_fn(unused_i): function
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 259 auto body_fn = [&](XlaOp j, absl::Span<const XlaOp> values, in ComputeWYRepresentation() local
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 885 def body_fn(t, state, ta): function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 734 NameAttrList body_fn; in TEST_F() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2420 tensorflow::ops::BodyGraphBuilderFn body_fn = in TF_FinishWhileHelper() local
|