| /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/transforms/ |
| D | resource_device_inference.cc | 71 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult()
149 LogicalResult ComputeResourceDevicesInComputation(FuncOp func_op, in ComputeResourceDevicesInComputation()
240 for (auto func_op : module.getOps<FuncOp>()) { in runOnOperation() local
275 auto func_op = worklist.pop_back_val(); in runOnOperation() local
|
| D | tpu_update_embedding_enqueue_op_inputs.cc | 64 FuncOp func_op, llvm::StringMap<Operation*>* enqueue_op_map, in FindTPUEmbeddingOps()
142 auto func_op = getFunction(); in runOnFunction() local
|
| D | cluster_tf_ops_pass.cc | 111 FuncOp func_op) { in GetFunctionMetadatas()
208 FuncOp func_op = FuncOp::create(loc, func_name, func_type); in CreateFunctions() local
306 for (auto func_op : module_op.getOps<FuncOp>()) { in runOnOperation() local
309 for (auto func_op : original_func) { in runOnOperation() local
|
| D | cross_host_transfer.cc | 73 module.walk([&](FuncOp func_op) { runOnFunction(func_op, key_count); }); in runOnOperation()
76 void CrossHostTransferPass::runOnFunction(FuncOp func_op, int &key_count) { in runOnFunction()
|
| D | tensor_device_copy_conversion.cc | 42 FuncOp func_op = getFunction(); in runOnFunction() local
|
| D | executor_tpuv1_inline_tpu_island.cc | 82 for (FuncOp func_op : in runOnOperation() local
|
| D | resource_op_lifting.cc | 682 FuncOp func_op, llvm::SmallDenseMap<int64_t, ResourceArgUseInfo>* result) { in FindResourceArgUseInfo()
749 FuncOp func_op, in RemoveUnusedResourceArgumentsAndForwardedRetvals()
799 FuncOp func_op, in LiftArgRetResourcesForFunction()
1311 auto walk_result = module.walk([&](FuncOp func_op) { in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | side_effect_analysis.h | 43 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo()
|
| D | side_effect_analysis.cc | 295 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in AnalyzeFunction()
|
| D | resource_alias_analysis.cc | 244 FuncOp func_op, const BacktrackAnalysis& backtrack_analysis) { in ResourceAliasAnalysisInfo()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1310 tensorflow::Node* func_op; in TEST_F() local
1336 tensorflow::Node* func_op; in TEST_F() local
1370 tensorflow::Node* func_op; in TEST_F() local
1404 tensorflow::Node* func_op; in TEST_F() local
1465 tensorflow::Node* func_op; in TEST_F() local
1526 tensorflow::Node* func_op; in TEST_F() local
2321 tensorflow::Node* func_op; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_saved_model.cc | 482 for (auto func_op : to_remove) rewriter.eraseOp(func_op); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 731 auto func_op = in GetAllocationIndex() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 980 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc() local
1050 const auto func_op = GetFunc(ctx, default_value.func(), status); in SetOpAttrValueScalar() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 628 LogicalResult ConvertKerasLSTMLayer(mlir::FuncOp func_op, OpBuilder* builder) { in ConvertKerasLSTMLayer()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1441 auto func_op = FuncOp::create(builder_.getUnknownLoc(), function_name, in Initialize() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3664 auto func_op = sub_symbol_table.lookup<mlir::FuncOp>(sig_def_key); in ConvertSignature() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 529 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc() local
|