| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | loops.cc | 27 const WhileLoopHelperBodyFunction& body_function, in WhileLoopHelper() argument
69 body_function(unpack_tuple(parameter, arity, body_builder.get()), in WhileLoopHelper()
84 const ForEachIndexBodyFunction& body_function, in ForEachIndex() argument
105 body_function(iteration, values, body_builder)); in ForEachIndex()
|
| D | loops.h | 51 const WhileLoopHelperBodyFunction& body_function,
66 const ForEachIndexBodyFunction& body_function,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf_while.cc | 68 create_region_with_call(while_op.body_function(), new_op.body()); in RunOnWhile()
|
| D | lower_static_tensor_list.cc | 362 auto inside_uses = while_op.body_function() in matchAndRewrite()
807 for (FuncOp func : {op.cond_function(), op.body_function()}) { in GetTensorListArgumentsFromWhileOp()
839 for (FuncOp func : {op.cond_function(), op.body_function()}) { in UpdateFunctionTypes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | resource_op_lifting_cleanup.cc | 198 FuncOp body = op.body_function(); in EliminateUnusedResultsForWhile()
313 for (FuncOp func : {op.cond_function(), op.body_function()}) { in CanonicalizeFunctionalWhile()
319 if (failed(ForwardCommonArgToOutput(op, {op.body_function()}, in CanonicalizeFunctionalWhile()
|
| D | functional_control_flow_to_regions.cc | 131 CreateCall(while_op, while_op.body_function(), in ConvertWhileOp()
|
| D | optimize_global_tensors.cc | 118 {while_op.cond_function(), while_op.body_function()}) { in AnalyzeFunc()
|
| D | resource_device_inference.cc | 286 {while_op.body_function(), while_op.cond_function()}, in runOnOperation()
|
| D | functional_control_flow_to_cfg.cc | 195 auto body_fn = op.body_function(); in LowerWhileOp()
|
| D | shape_inference.cc | 1075 while_op.body_function().getType().getResults()); in InferShapeForSingleOperation()
1343 while_op.body_function().getType().getInputs()); in PropagateShapeIntoAttachedFunctions()
1346 {while_op.cond_function(), while_op.body_function()}, max_iteration); in PropagateShapeIntoAttachedFunctions()
1350 {while_op.cond_function(), while_op.body_function()}, max_iteration); in PropagateShapeIntoAttachedFunctions()
|
| D | collection_ops_util.cc | 177 auto body = while_op.body_function(); in GetElementTypeFromAccess()
|
| D | tensor_array_ops_decomposition.cc | 494 {while_op.body_function(), while_op.cond_function()}, module)) in AccessedGradients()
560 auto body = while_op.body_function(); in HandleWhileOp()
|
| D | stack_ops_decomposition.cc | 164 auto body = while_op.body_function(); in HandleWhileOp()
|
| D | tensor_list_ops_decomposition.cc | 169 auto body = while_op.body_function(); in HandleWhileOp()
|
| D | resource_op_lifting.cc | 1238 auto body = while_op.body_function(); in HoistForControlFlow()
|
| /external/python/pycparser/pycparser/ |
| D | c_generator.py | 309 body_function = self._generate_struct_union_body
313 body_function = self._generate_enum_body
322 s += body_function(members)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 318 while_op.body_function())); in ResourceAliasAnalysisInfo()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 167 ImportXlaRegion(op.body_function(), &while_op.body(), loc); in LowerWhile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2896 auto body_fn = op.body_function(); in Verify()
|
| D | tf_ops.td | 732 FuncOp body_function() {
|