Home
last modified time | relevance | path

Searched refs:FunctionDefHelper (Results 1 – 25 of 81) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dparallel_filter_dataset_op_test.cc34 FunctionDefHelper::AttrValueWrapper pred_func, in ParallelFilterDatasetParams()
90 FunctionDefHelper::AttrValueWrapper pred_func_;
108 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in ParallelFilterDatasetParams1()
127 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in ParallelFilterDatasetParams2()
146 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in ParallelFilterDatasetParams3()
165 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in ParallelFilterDatasetParams4()
184 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in ParallelFilterDatasetParams5()
203 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in ParallelFilterDatasetParams6()
222 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in InputHasNoElementParams()
242 FunctionDefHelper::FunctionRef("GetUnique", in InvalidPredFuncFilterDatasetParams1()
[all …]
H A Dmap_defun_op_test.cc30 FunctionDefHelper::AttrValueWrapper func, in MapDefunOpParams()
85 FunctionDefHelper::AttrValueWrapper func_;
133 {FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_INT64}})}, in TestCase1()
155 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in TestCase2()
175 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in TestCase3()
195 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in InvalidOutputTypes()
213 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in InvalidOutputShapes()
232 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in InvalidInputs()
H A Dfilter_dataset_op_test.cc27 FunctionDefHelper::AttrValueWrapper pred_func, in FilterDatasetParams()
77 FunctionDefHelper::AttrValueWrapper pred_func_;
93 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in FilterDatasetParams1()
110 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in FilterDatasetParams2()
128 FunctionDefHelper::FunctionRef("GetUnique", in InvalidPredFuncFilterDatasetParams1()
148 FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in InvalidPredFuncFilterDatasetParams2()
166 FunctionDefHelper::FunctionRef("NonZero", {{"T", DT_INT64}}), in InvalidPredFuncFilterDatasetParams3()
H A Dreduce_dataset_op_test.cc29 FunctionDefHelper::AttrValueWrapper func, in ReduceDatasetParams()
88 FunctionDefHelper::AttrValueWrapper func_;
103 /*func=*/FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}}), in ReduceDatasetParams1()
125 FunctionDefHelper::FunctionRef("XPlusOneXTimesY", {{"T", DT_INT64}}), in ReduceDatasetParams2()
143 FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}}), in ReduceDatasetParams3()
H A Dmap_dataset_op_test.cc32 FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_INT64}}), in MapDatasetParams1()
44 FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_INT64}}), in MapDatasetParams1()
67 FunctionDefHelper::FunctionRef("XAddX", {{"T", DT_INT64}}), in MapDatasetParams2()
84 FunctionDefHelper::FunctionRef("XTimesFour", {{"T", DT_INT64}}), in MapDatasetParams3()
H A Dflat_map_dataset_op_test.cc27 FunctionDefHelper::AttrValueWrapper func, in FlatMapDatasetParams()
75 FunctionDefHelper::AttrValueWrapper func_;
88 auto func = FunctionDefHelper::FunctionRef( in FlatMapDatasetParams1()
111 auto func = FunctionDefHelper::FunctionRef(/*name*/ "NonZero", in InvalidFlatMapDatasetParams()
H A Dparallel_map_dataset_op_test.cc29 int num_parallel_calls, FunctionDefHelper::AttrValueWrapper func, in ParallelMapDatasetParams()
92 FunctionDefHelper::AttrValueWrapper func_;
102 FunctionDefHelper::AttrValueWrapper MapFunc(const string& func_name, in MapFunc()
104 return FunctionDefHelper::FunctionRef(func_name, {{"T", dtype}}); in MapFunc()
H A Dinterleave_dataset_op_test.cc28 FunctionDefHelper::AttrValueWrapper func, in InterleaveDatasetParams()
90 FunctionDefHelper::AttrValueWrapper func_;
97 FunctionDefHelper::AttrValueWrapper MakeTensorSliceDatasetFunc( in MakeTensorSliceDatasetFunc()
100 return FunctionDefHelper::FunctionRef( in MakeTensorSliceDatasetFunc()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_optimizer_test.cc92 FunctionDef x_times_two = FunctionDefHelper::Define( in TEST_F()
148 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
197 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
257 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
304 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
310 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
350 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
354 FunctionDefHelper::Const("zero", 0), in TEST_F()
355 FunctionDefHelper::Const("one", 1), in TEST_F()
399 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
[all …]
H A Dmeta_optimizer_test.cc281 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
287 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
294 FunctionDef quadratic_func = FunctionDefHelper::Create( in TEST_F()
430 FunctionDef my_mul = FunctionDefHelper::Create( in TEST_F()
439 FunctionDef my_fwd = FunctionDefHelper::Create( in TEST_F()
482 FunctionDef expected_my_mul = FunctionDefHelper::Create( in TEST_F()
489 FunctionDef expected_my_fwd = FunctionDefHelper::Create( in TEST_F()
532 FunctionDef my_func = FunctionDefHelper::Create( in TEST_F()
622 using FDH = FunctionDefHelper; in TEST_F()
642 FunctionDef mul_func_1 = FunctionDefHelper::Create( in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dfunctions_test.cc39 FunctionDef parametrized_func = FunctionDefHelper::Create( in TEST_F()
46 FunctionDef non_parametrized_func = FunctionDefHelper::Create( in TEST_F()
63 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
110 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
177 std::vector<FunctionDefHelper::Node> nodes = { in TEST_F()
195 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
269 TF_ASSERT_OK(flib.AddFunctionDef(FunctionDefHelper::Define( in TEST_F()
282 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
353 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
406 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils_test.cc92 *f1 = FunctionDefHelper::Create( in TEST_F()
100 *f2 = FunctionDefHelper::Create( in TEST_F()
115 *f1 = FunctionDefHelper::Create( in TEST_F()
123 *f2 = FunctionDefHelper::Create( in TEST_F()
141 *f1 = FunctionDefHelper::Create( in TEST_F()
149 *f2 = FunctionDefHelper::Create( in TEST_F()
484 *f1 = FunctionDefHelper::Create( in TEST_F()
492 *f2 = FunctionDefHelper::Create( in TEST_F()
548 *f1 = FunctionDefHelper::Create( in TEST_F()
556 *f2 = FunctionDefHelper::Create( in TEST_F()
[all …]
H A Ddataset_utils_test.cc52 return FunctionDefHelper::Create( in TEST()
62 return FunctionDefHelper::Create( in TEST()
100 *fdef_base.add_function() = FunctionDefHelper::Create( in TEST()
109 *fdef_to_add.add_function() = FunctionDefHelper::Create( in TEST()
137 *flib.add_function() = FunctionDefHelper::Create( in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dcompilability_check_util_test.cc181 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
222 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
232 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
302 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
309 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
403 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
410 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
457 FunctionDef identity_func = FunctionDefHelper::Create( in TEST_F()
485 FunctionDef sort_func = FunctionDefHelper::Create( in TEST_F()
516 FunctionDef identity_func = FunctionDefHelper::Create( in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dsplit_utils_test.cc189 FunctionDef orig = FunctionDefHelper::Create( in TEST()
234 FunctionDef orig = FunctionDefHelper::Create( in TEST()
263 FunctionDef orig = FunctionDefHelper::Create( in TEST()
297 FunctionDef orig = FunctionDefHelper::Create( in TEST()
331 FunctionDef orig = FunctionDefHelper::Create( in TEST()
374 FunctionDefHelper::Create("MyFunction", // function_name in TEST()
400 FunctionDef orig = FunctionDefHelper::Create( in TEST()
433 FunctionDef orig = FunctionDefHelper::Create( in TEST()
452 orig = FunctionDefHelper::Create( in TEST()
475 FunctionDef orig = FunctionDefHelper::Create( in TEST()
H A Dmake_deterministic_test.cc55 FunctionDef orig_func_def = FunctionDefHelper::Create( in TEST_P()
215 return FunctionDefHelper::Define( in OuterXTimesTwo()
230 FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_FLOAT}})}}}}); in OuterXTimesTwo()
235 return FunctionDefHelper::Define( in OuterRandomUniform()
249 {"f", FunctionDefHelper::FunctionRef("RandomUniformFn", in OuterRandomUniform()
255 return FunctionDefHelper::Define( in OuterReadResourceVariable()
269 {"f", FunctionDefHelper::FunctionRef("ReadResourceVariable", {})}}}}); in OuterReadResourceVariable()
340 typedef FunctionDefHelper FDH; in TEST_P()
374 typedef FunctionDefHelper FDH; in TEST_P()
413 typedef FunctionDefHelper FDH; in TEST_P()
[all …]
H A Dgraph_test_utils.cc75 {{"predicate", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeFilterNode()
90 {{"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeMapAndBatchNode()
100 {{"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeMapNode()
117 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelInterleaveV2Node()
137 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelInterleaveV4Node()
152 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelMapNode()
168 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelMapV2Node()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction_testlib.h37 std::pair<string, FunctionDefHelper::AttrValueWrapper>>& attrs) { in Attrs()
44 const std::vector<std::pair<string, FunctionDefHelper::AttrValueWrapper>>& in Attrs()
60 gtl::ArraySlice<std::pair<string, FunctionDefHelper::AttrValueWrapper>>
H A Dgraph_to_functiondef_test.cc75 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
112 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
152 FunctionDefHelper::Create("test_fn", // function name in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate_data.cc72 std::vector<tensorflow::FunctionDefHelper::Node> nodes; in BuildFunctionDefProto()
75 nodes.push_back(tensorflow::FunctionDefHelper::Const<tensorflow::tstring>( in BuildFunctionDefProto()
78 tensorflow::FunctionDefHelper::Node execute_node; in BuildFunctionDefProto()
93 fdef = tensorflow::FunctionDefHelper::Create(function_name, inputs, outputs, in BuildFunctionDefProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dio_ops_test.cc39 FunctionDefHelper::AttrValueWrapper shard_func, in SaveDatasetV2Params()
94 FunctionDefHelper::AttrValueWrapper shard_func_;
120 FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_INT64}}), in SaveDatasetV2Params1()
136 FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_INT64}}), in SaveDatasetV2Params2()
H A Dmap_and_batch_dataset_op_test.cc29 FunctionDefHelper::AttrValueWrapper func, in MapAndBatchDatasetParams()
94 FunctionDefHelper::AttrValueWrapper func_;
102 FunctionDefHelper::AttrValueWrapper MapFunc(const string& func_name, in MapFunc()
104 return FunctionDefHelper::FunctionRef(func_name, {{"T", dtype}}); in MapFunc()
H A Dparallel_interleave_dataset_op_test.cc32 int64_t prefetch_input_elements, FunctionDefHelper::AttrValueWrapper func, in ParallelInterleaveDatasetParams()
107 FunctionDefHelper::AttrValueWrapper func_;
114 FunctionDefHelper::AttrValueWrapper MakeTensorSliceDatasetFunc( in MakeTensorSliceDatasetFunc()
117 return FunctionDefHelper::FunctionRef( in MakeTensorSliceDatasetFunc()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state_test.cc342 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
399 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
447 auto pow3_fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
454 auto base2pow3_fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
463 {"f", tensorflow::FunctionDefHelper::FunctionRef( in TEST_F()
519 auto pow3_fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
536 auto base2pow3_fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
545 {"f", tensorflow::FunctionDefHelper::FunctionRef( in TEST_F()
584 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
651 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libtf/tests/
H A Dfunction_test.cc39 using tensorflow::FunctionDefHelper;
80 return FunctionDefHelper::Define( in SquareFunc()
100 return FunctionDefHelper::Define( in AddFunc()
120 return FunctionDefHelper::Define( in IdentityNFunc()

1234