Home
last modified time | relevance | path

Searched refs:branch_func (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc53 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode() local
54 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
55 (*branch_func.mutable_attr())["_device_ordinal"] = device_ordinal_value; in SetDeviceOrdinalAttributeForNode()
57 node->AddAttr(attr_name, branch_func); in SetDeviceOrdinalAttributeForNode()
63 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode() local
64 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
65 (*branch_func.mutable_attr())["_device_ordinal"] = device_ordinal_value; in SetDeviceOrdinalAttributeForNode()
67 node->AddAttr(attr_name, branch_func); in SetDeviceOrdinalAttributeForNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc345 FunctionDef* branch_func = library->add_function(); in AddBranch() local
346 auto* signature = branch_func->mutable_signature(); in AddBranch()
347 (*branch_func->mutable_attr())[data::kTFDataFunction].set_b(true); in AddBranch()
348 graph_utils::SetUniqueGraphFunctionName("branch", library, branch_func); in AddBranch()
368 auto function_node = branch_func->add_node_def(); in AddBranch()
370 function_utils::SetUniqueFunctionNodeName(node->name(), branch_func, in AddBranch()
404 (*branch_func->mutable_ret())["output"] = prev_node_output; in AddBranch()
408 func_attr.set_name(branch_func->signature().name()); in AddBranch()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc643 Type result_type, FuncOp branch_func, in CreateCondTrueBranch()
647 rewriter->createBlock(&branch_func.getBody(), branch_func.begin(), in CreateCondTrueBranch()
648 branch_func.getType().getInputs()); in CreateCondTrueBranch()
680 FuncOp branch_func, in CreateCondFalseBranch()
687 rewriter->createBlock(&branch_func.getBody(), branch_func.begin(), in CreateCondFalseBranch()
688 branch_func.getType().getInputs()); in CreateCondFalseBranch()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc142 mlir::FuncOp branch_func = op.branch_function(i); in LowerCase() local
143 ImportXlaRegion(branch_func, &case_op.branches()[i], loc, in LowerCase()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc412 NameAttrList branch_func; in ResetDeviceOrdinalToPlaceholderValue() local
413 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), attr_name, &branch_func)); in ResetDeviceOrdinalToPlaceholderValue()
414 (*branch_func.mutable_attr())["_device_ordinal"] = device_ordinal_value; in ResetDeviceOrdinalToPlaceholderValue()
416 n->AddAttr(attr_name, branch_func); in ResetDeviceOrdinalToPlaceholderValue()
420 NameAttrList branch_func; in ResetDeviceOrdinalToPlaceholderValue() local
421 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), attr_name, &branch_func)); in ResetDeviceOrdinalToPlaceholderValue()
422 (*branch_func.mutable_attr())["_device_ordinal"] = device_ordinal_value; in ResetDeviceOrdinalToPlaceholderValue()
424 n->AddAttr(attr_name, branch_func); in ResetDeviceOrdinalToPlaceholderValue()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc782 auto branch_func = SymbolTable::lookupNearestSymbolFrom<FuncOp>( in VerifyCaseOrIfOpBranchFunctions() local
784 if (!branch_func) in VerifyCaseOrIfOpBranchFunctions()
789 FunctionType branch_type = branch_func.getType(); in VerifyCaseOrIfOpBranchFunctions()