Searched refs:branch_func (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | side_effect_util.cc | 53 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/ |
D | map_vectorization.cc | 345 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/ |
D | lower_static_tensor_list.cc | 643 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/ |
D | legalize_tf_control_flow.cc | 142 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/ |
D | extract_outside_compilation_pass.cc | 412 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/ |
D | tf_ops_a_m.cc | 782 auto branch_func = SymbolTable::lookupNearestSymbolFrom<FuncOp>( in VerifyCaseOrIfOpBranchFunctions() local 784 if (!branch_func) in VerifyCaseOrIfOpBranchFunctions() 789 FunctionType branch_type = branch_func.getType(); in VerifyCaseOrIfOpBranchFunctions()
|