Searched refs:body_attr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 514 const AttrValue* body_attr = n->attrs().Find("body"); in RewriteWhileNode() local
515 if (body_attr == nullptr) { in RewriteWhileNode()
525 n, cond_attr->func(), body_attr->func(), parallel_iterations_attr->i(), g, in RewriteWhileNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 258 NameAttrList body_attr; in PropagateConstIntoWhileNode() local
259 TF_RETURN_IF_ERROR(GetNodeAttr(while_node->def(), "body", &body_attr)); in PropagateConstIntoWhileNode()
260 const FunctionDef* body_func = lookup_fld->Find(body_attr.name()); in PropagateConstIntoWhileNode()
262 return errors::Internal("Cannot find body function ", body_attr.name(), in PropagateConstIntoWhileNode()
285 body_attr.name()); in PropagateConstIntoWhileNode()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 590 auto body_attr = builder.getSymbolRefAttr(func_names.at(body_idx)); in ConvertSubgraphIdxsToFunctionAttrs() local
593 builder.getNamedAttr("body", body_attr)}; in ConvertSubgraphIdxsToFunctionAttrs()
|