Home
last modified time | relevance | path

Searched defs:branch_func (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc53 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode() local
63 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc142 mlir::FuncOp branch_func = op.branch_function(i); in LowerCase() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc345 FunctionDef* branch_func = library->add_function(); in AddBranch() local
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc412 NameAttrList branch_func; in ResetDeviceOrdinalToPlaceholderValue() local
420 NameAttrList branch_func; in ResetDeviceOrdinalToPlaceholderValue() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc782 auto branch_func = SymbolTable::lookupNearestSymbolFrom<FuncOp>( in VerifyCaseOrIfOpBranchFunctions() local