Searched defs:func_op (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | resource_device_inference.cc | 63 explicit PerFunctionResult(FuncOp func_op) : alias_analysis_(func_op) {} in PerFunctionResult()
125 LogicalResult ComputeResourceDevicesInComputation(FuncOp func_op, in ComputeResourceDevicesInComputation()
197 module.walk([&](FuncOp func_op) { in runOnModule()
236 auto func_op = worklist.back(); in runOnModule() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function_test.cc | 410 TF_Operation* func_op = Use({}); in TEST_F() local
431 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
452 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
473 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
492 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
517 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
542 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
565 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
617 TF_Operation* func_op = Use({two, ten, func_feed}); in TEST_F() local
648 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | side_effect_analysis.cc | 85 FuncOp func_op) { in FindPassthroughArgumentForReturnValue()
116 auto func_op = llvm::dyn_cast<FuncOp>(op); in ResourceAliasAnalysis() local
121 void ResourceAliasAnalysis::AnalyzeFunction(FuncOp func_op) { in AnalyzeFunction()
401 FuncOp func_op, const ResourceAliasAnalysis& alias_analysis) { in AnalyzeFunction()
543 auto func_op = llvm::dyn_cast<FuncOp>(op); in SideEffectAnalysis() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 569 mlir::FuncOp* func_op) { in UpdateFuncSignature()
579 LogicalResult ConvertKerasLSTMLayer(mlir::FuncOp func_op, OpBuilder* builder) { in ConvertKerasLSTMLayer()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1148 tensorflow::Node* func_op; in TEST_F() local
1174 tensorflow::Node* func_op; in TEST_F() local
1208 tensorflow::Node* func_op; in TEST_F() local
1242 tensorflow::Node* func_op; in TEST_F() local
1304 tensorflow::Node* func_op; in TEST_F() local
2024 tensorflow::Node* func_op; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | extract_ophint.cc | 537 FuncOp func_op = FuncOp::create(insert_before_op->getLoc(), func_name, in BuildFusedFuncOp() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 1578 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc() local
1629 const auto func_op = GetFunc(ctx, default_value.func(), status); in SetOpAttrValueScalar() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 2956 auto func_op = symbol_table.lookup<mlir::FuncOp>(sig_def_key); in ConvertSignature() local
3014 auto func_op = op.getParentOfType<mlir::FuncOp>(); in LiftVariable() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 510 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc() local
|