Searched refs:BuildFunction (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_outlining.cc | 58 FuncOp BuildFunction(llvm::ArrayRef<Value> live_ins, ClusterOrLaunchOp op, in BuildFunction() function
109 BuildFunction(live_ins.getArrayRef(), cluster_op, symbol_table, builder); in OutlineCluster()
131 BuildFunction(live_ins.getArrayRef(), launch_op, symbol_table, builder); in OutlineLaunch()
|
| D | tpu_extract_outside_compilation.cc | 67 FuncOp BuildFunction(llvm::ArrayRef<Operation*> ops, in BuildFunction() function
532 FuncOp shape_op = BuildFunction(clustered_ops.getArrayRef(), in MoveOpsToHost()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | unified_api_testutil.cc | 30 AbstractContext* BuildFunction(const char* fn_name) { in BuildFunction() function
64 AbstractContextPtr func_ctx(BuildFunction(fn_name)); in RunModel()
|
| D | unified_api_test.cc | 144 AbstractContextPtr ctx(BuildFunction("test_fn")); in TEST_P()
168 AbstractContextPtr ctx(BuildFunction("test_fn")); in TEST_P()
|
| D | unified_api_testutil.h | 30 AbstractContext* BuildFunction(const char* fn_name);
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | function_cache_test.cc | 93 tensorflow::AbstractContext* BuildFunction(const char* fn_name) { in BuildFunction() function
127 tensorflow::AbstractContextPtr func_ctx(BuildFunction(kFunctionName)); in PrepareFunction()
|