Home
last modified time | relevance | path

Searched defs:symbol_ref (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_outline_tpu_island.cc55 void MoveFuncOp(FlatSymbolRefAttr &symbol_ref, SymbolTable &from, in MoveFuncOp()
180 if (auto symbol_ref = attr.getValue().dyn_cast<FlatSymbolRefAttr>()) { in runOnOperation() local
186 auto symbol_ref = attribute.dyn_cast<FlatSymbolRefAttr>(); in runOnOperation() local
Dexecutor_tpuv1_island_coarsening.cc625 auto symbol_ref = attr.getValue().dyn_cast<FlatSymbolRefAttr>(); in runOnOperation() local
/external/tensorflow/tensorflow/dtensor/mlir/
Dpropagate_device_id_to_function_args.cc67 auto symbol_ref = llvm::dyn_cast<mlir::TF::PartitionedCallOp>(op).f(); in FindFunctionsToRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.cc390 if (auto symbol_ref = attr.dyn_cast<mlir::SymbolRefAttr>()) { in ConvertAttributes() local
/external/tensorflow/tensorflow/core/ir/importexport/
Dconvert_attributes.cc202 if (auto symbol_ref = attr.dyn_cast<SymbolRefAttr>()) { in ConvertAttribute() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3416 auto symbol_ref = mlir::SymbolRefAttr::get( in CreateSavedModelIR() local