Home
last modified time | relevance | path

Searched defs:func_op (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dreorder_assert.cc38 for (auto func_op : module.getOps<mlir::FuncOp>()) { in runOnOperation() local
43 void ProcessFunction(mlir::FuncOp func_op) { in ProcessFunction()
80 bool IsFunctionNonSideEffectingOrAssert(mlir::FuncOp func_op) { in IsFunctionNonSideEffectingOrAssert()
Dcorert_converter.cc171 FuncOp func_op = op->getParentOfType<mlir::FuncOp>(); in ConvertOpHandler() local
182 mlir::FuncOp func_op = op->getParentOfType<mlir::FuncOp>(); in GetDistributedContext() local
199 mlir::FuncOp func_op = op->getParentOfType<mlir::FuncOp>(); in GetRemoteChainManager() local
220 auto func_op = op->getParentOfType<mlir::FuncOp>(); in GetLocalSideEffectChain() local
252 mlir::FuncOp func_op = op->getParentOfType<mlir::FuncOp>(); in GetTaskHandle() local
Dset_shape_invariant_in_while_ops.cc28 mlir::FuncOp func_op = getOperation(); in runOnOperation() local
Doptimize_tf_control_flow_side_effect.cc25 mlir::FuncOp func_op, in FunctionHasSideEffect()
Dinsert_tensor_copy.cc46 mlir::FuncOp func_op = getOperation(); in runOnOperation() local
Dcross_device_transfer.cc123 FuncOp func_op = getOperation(); in runOnFunction() local
Dremove_tf_if_const_args.cc40 for (auto func_op : in runOnOperation() local
Dtf_to_tfrt.cc93 auto func_op = op->getParentOfType<mlir::FuncOp>(); in GetFunctionInputChain() local
638 mlir::FuncOp func_op, llvm::ArrayRef<Value> operands, in matchAndRewrite()
1276 auto func_op = op->getParentOfType<mlir::FuncOp>(); in GetPredicateFunction() local
1337 auto func_op = op->getParentOfType<mlir::FuncOp>(); in GetWhileBodyFunction() local
1675 void ChainDanglingValuesinFunction(mlir::FuncOp func_op) { in ChainDanglingValuesinFunction()
1721 auto func_op = builder.create<mlir::FuncOp>( in CreateFallbackInitializationFunction() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/
Dtensor_array_side_effect_analysis.cc31 mlir::FuncOp func_op) { in FunctionContainsOnlyNoSideEffectOpOrTensorArrayOp()
43 for (auto func_op : module.getOps<mlir::FuncOp>()) { in TensorArraySideEffectAnalysis() local
Dtensor_array_side_effect_analysis.h38 bool HasAtMostTensorArrayEffect(mlir::FuncOp func_op) const { in HasAtMostTensorArrayEffect()
Dtest_cost_analysis_pass.cc27 auto func_op = getOperation(); in runOnOperation() local
Dtest_tensor_array_side_effect_analysis.cc29 for (auto func_op : module.getOps<mlir::FuncOp>()) { in runOnOperation() local
Dcost_analysis.h36 explicit CostAnalysis(mlir::FuncOp func_op) { in CostAnalysis()
/external/tensorflow/tensorflow/c/
Dc_api_function_test.cc410 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/transforms/
Dresource_device_inference.cc78 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult()
156 LogicalResult ComputeResourceDevicesInComputation(FuncOp func_op, in ComputeResourceDevicesInComputation()
247 for (auto func_op : module.getOps<FuncOp>()) { in runOnOperation() local
282 auto func_op = worklist.pop_back_val(); in runOnOperation() local
Dcluster_tf_ops_pass.cc111 FuncOp func_op) { in GetFunctionMetadatas()
208 FuncOp func_op = FuncOp::create(loc, func_name, func_type); in CreateFunctions() local
313 for (auto func_op : module_op.getOps<FuncOp>()) { in runOnOperation() local
316 for (auto func_op : original_func) { in runOnOperation() local
Dtpu_update_embedding_enqueue_op_inputs.cc66 FuncOp func_op, llvm::StringMap<Operation*>* enqueue_op_map, in FindTPUEmbeddingOps()
144 auto func_op = getFunction(); in runOnFunction() local
Dmark_initialized_variables_pass.cc59 auto func_op = getFunction(); in runOnFunction() local
Dexecutor_tpuv1_inline_tpu_island.cc82 for (FuncOp func_op : in runOnOperation() local
Dtensor_device_copy_conversion.cc49 FuncOp func_op = getFunction(); in runOnFunction() local
Dinitialize_variables_in_session_init.cc141 for (auto func_op : module.getOps<FuncOp>()) { in runOnOperation() local
Dresource_op_lifting.cc611 FuncOp func_op, llvm::SmallDenseMap<int64_t, ResourceArgUseInfo>* result) { in FindResourceArgUseInfo()
678 FuncOp func_op, in RemoveUnusedResourceArgumentsAndForwardedRetvals()
728 FuncOp func_op, in LiftArgRetResourcesForFunction()
1240 auto walk_result = module.walk([&](FuncOp func_op) { in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dside_effect_analysis.h43 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo()
/external/tensorflow/tensorflow/core/ir/
Dops.cc81 auto func_op = dyn_cast<GraphFuncOp>(block->getParentOp()); in getAsmBlockArgumentNames() local
468 LogicalResult GraphFuncOp::canonicalize(GraphFuncOp func_op, in canonicalize()
490 static LogicalResult verifyGraphFunc(GraphFuncOp func_op) { in verifyGraphFunc()
/external/tensorflow/tensorflow/core/transforms/graph_to_func/
Dgraph_to_func.cc62 auto func_op = builder.create<GraphFuncOp>(graph.getLoc(), name, func_type, in GraphToFunc() local

12