Home
last modified time | relevance | path

Searched defs:case_op (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc137 auto case_op = in LowerCase() local
305 auto case_op = in LowerCaseRegion() local
394 if (auto case_op = dyn_cast<TF::CaseOp>(op)) { in runOnOperation() local
Dmhlo_to_lhlo_with_xla.cc1332 auto case_op = in EmitCaseOp() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_case_op.cc98 CaseBuilder::CaseBuilder(Node* case_op, in CaseBuilder()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc372 TF::CaseRegionOp case_op, ModuleOp module, in HandleCaseRegionOp()
883 } else if (auto case_op = llvm::dyn_cast<TF::CaseOp>(&op)) { in DecomposeTensorListOpsInternal() local
918 } else if (auto case_op = llvm::dyn_cast<TF::CaseRegionOp>(&op)) { in DecomposeTensorListOpsInternal() local
Dresource_op_lifting_cleanup.cc423 } else if (auto case_op = dyn_cast<TF::CaseOp>(op)) { in CleanupAndCanonicalize() local
Dresource_op_lifting.cc1185 } else if (auto case_op = llvm::dyn_cast<TF::CaseOp>(&op)) { in HoistForControlFlow() local
Dshape_inference.cc1557 } else if (auto case_op = dyn_cast<TF::CaseOp>(op)) { in PropagateShapeIntoAttachedFunctions() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc367 } else if (auto case_op = dyn_cast<CaseOp>(op)) { in ResourceAliasAnalysisInfo() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dlower_saved_model.cc678 } else if (auto case_op = llvm::dyn_cast<mlir::TF::CaseOp>(op)) { in PromoteOpOperand() local