Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc164 auto case_op = builder.create<mhlo::CaseOp>( in LowerCase() local
346 auto case_op = builder.create<mhlo::CaseOp>( in LowerCaseRegion() local
438 if (auto case_op = dyn_cast<TF::CaseOp>(op)) { in runOnOperation() local
Dmhlo_to_lhlo_with_xla.cc1386 auto case_op = in EmitCaseOp() local
/external/tensorflow/tensorflow/core/ir/
Dtf_op_registry_test.cc104 Operation *case_op = in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_case_op.cc98 CaseBuilder::CaseBuilder(Node* case_op, in CaseBuilder()
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dsave_restore_spmd_expander.cc372 auto case_op = builder.create<mlir::TF::CaseOp>( in ConditionalSave() local
645 auto case_op = builder.create<mlir::TF::CaseOp>( in ExpandRestoreV2OpHelper() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc375 TF::CaseRegionOp case_op, ModuleOp module, in HandleCaseRegionOp()
889 } else if (auto case_op = llvm::dyn_cast<TF::CaseOp>(&op)) { in DecomposeTensorListOpsInternal() local
924 } else if (auto case_op = llvm::dyn_cast<TF::CaseRegionOp>(&op)) { in DecomposeTensorListOpsInternal() local
Dresource_op_lifting_cleanup.cc427 } else if (auto case_op = dyn_cast<TF::CaseOp>(op)) { in CleanupAndCanonicalize() local
Dresource_op_lifting.cc1192 } else if (auto case_op = llvm::dyn_cast<TF::CaseOp>(&op)) { in HoistForControlFlow() local
Dshape_inference.cc2382 } else if (auto case_op = dyn_cast<TF::CaseOp>(op)) { in PropagateShapeIntoAttachedFunctions() local
/external/skia/src/sksl/codegen/
DSkSLRasterPipelineBuilder.h424 void case_op(int value) { in case_op() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc388 } else if (auto case_op = dyn_cast<CaseOp>(op)) { in ResourceAliasAnalysisInfo() local
Dside_effect_analysis.cc249 } else if (auto case_op = dyn_cast<CaseOp>(op)) { in CollectOpSideEffects() local
/external/skia/src/opts/
DSkRasterPipeline_opts.h3285 STAGE_TAIL(case_op, SkRasterPipeline_CaseOpCtx* ctx) { in STAGE_TAIL() argument