Home
last modified time | relevance | path

Searched defs:branch_func (Results 1 – 4 of 4) 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.cc143 mlir::FuncOp branch_func = op.branch_function(i); in LowerCase() 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.cc778 auto branch_func = symbol_table.lookupNearestSymbolFrom<FuncOp>( in VerifyCaseOrIfOpBranchFunctions() local