| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils_test.cc | 110 FunctionDefHelper::Node Cast(string&& name, std::vector<string>&& inputs, in Cast()
159 FunctionDef inner = FunctionDefHelper::Create( in TEST()
234 FunctionDef inner = FunctionDefHelper::Create( in TEST()
312 FunctionDef inner = FunctionDefHelper::Create( in TEST()
386 FunctionDef inner = FunctionDefHelper::Create( in TEST()
458 FunctionDef inner = FunctionDefHelper::Create( in TEST()
547 FunctionDef inner = FunctionDefHelper::Create( in TEST()
553 {FunctionDefHelper::Const("Const", 2), in TEST()
623 FunctionDef inner = FunctionDefHelper::Create( in TEST()
629 {FunctionDefHelper::Const("Const", 2), in TEST()
[all …]
|
| D | graph_test_utils.cc | 60 {{"predicate", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeFilterNode()
75 {{"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeMapAndBatchNode()
85 {{"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeMapNode()
102 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelInterleaveV2Node()
117 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelMapNode()
|
| D | map_vectorization_test.cc | 73 *map_fn = FunctionDefHelper::Create( in AddMapFn()
94 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
106 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
357 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapAndBatchNode()
497 std::vector<std::pair<string, FunctionDefHelper::AttrValueWrapper>> attrs; in AddArbitraryInputNode()
|
| D | map_and_filter_fusion.cc | 118 FunctionDefHelper::FunctionRef(func->signature().name()).proto; in MakeFilterNode()
159 FunctionDefHelper::FunctionRef(func->signature().name()).proto; in MakeMapNode()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 39 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 …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | filter_dataset_op_test.cc | 27 FunctionDefHelper::AttrValueWrapper pred_func, in FilterDatasetParams()
76 FunctionDefHelper::AttrValueWrapper pred_func_;
92 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in FilterDatasetParams1()
109 /*pred_func=*/FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in FilterDatasetParams2()
127 FunctionDefHelper::FunctionRef("GetUnique", in InvalidPredFuncFilterDatasetParams1()
146 FunctionDefHelper::FunctionRef("IsZero", {{"T", DT_INT64}}), in InvalidPredFuncFilterDatasetParams2()
164 FunctionDefHelper::FunctionRef("NonZero", {{"T", DT_INT64}}), in InvalidPredFuncFilterDatasetParams3()
|
| D | map_defun_op_test.cc | 30 FunctionDefHelper::AttrValueWrapper func, in MapDefunOpParams()
85 FunctionDefHelper::AttrValueWrapper func_;
133 {FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_INT64}})}, in TestCase1()
154 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in TestCase2()
175 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in TestCase3()
194 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in InvalidOutputTypes()
212 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in InvalidOutputShapes()
231 {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})}, in InvalidInputs()
|
| D | reduce_dataset_op_test.cc | 29 FunctionDefHelper::AttrValueWrapper func, in ReduceDatasetParams()
87 FunctionDefHelper::AttrValueWrapper func_;
102 /*func=*/FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}}), in ReduceDatasetParams1()
124 FunctionDefHelper::FunctionRef("XPlusOneXTimesY", {{"T", DT_INT64}}), in ReduceDatasetParams2()
142 FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}}), in ReduceDatasetParams3()
|
| D | map_dataset_op_test.cc | 32 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()
|
| D | hash_utils_test.cc | 89 *f1 = FunctionDefHelper::Create( in TEST_F()
97 *f2 = FunctionDefHelper::Create( in TEST_F()
112 *f1 = FunctionDefHelper::Create( in TEST_F()
120 *f2 = FunctionDefHelper::Create( in TEST_F()
138 *f1 = FunctionDefHelper::Create( in TEST_F()
146 *f2 = FunctionDefHelper::Create( in TEST_F()
481 *f1 = FunctionDefHelper::Create( in TEST_F()
489 *f2 = FunctionDefHelper::Create( in TEST_F()
545 *f1 = FunctionDefHelper::Create( in TEST_F()
553 *f2 = FunctionDefHelper::Create( in TEST_F()
[all …]
|
| D | flat_map_dataset_op_test.cc | 27 FunctionDefHelper::AttrValueWrapper func, in FlatMapDatasetParams()
74 FunctionDefHelper::AttrValueWrapper func_;
87 auto func = FunctionDefHelper::FunctionRef( in FlatMapDatasetParams1()
110 auto func = FunctionDefHelper::FunctionRef(/*name*/ "NonZero", in InvalidFlatMapDatasetParams()
|
| D | parallel_map_dataset_op_test.cc | 29 int num_parallel_calls, FunctionDefHelper::AttrValueWrapper func, in ParallelMapDatasetParams()
93 FunctionDefHelper::AttrValueWrapper func_;
103 FunctionDefHelper::AttrValueWrapper MapFunc(const string& func_name, in MapFunc()
105 return FunctionDefHelper::FunctionRef(func_name, {{"T", dtype}}); in MapFunc()
|
| D | interleave_dataset_op_test.cc | 28 FunctionDefHelper::AttrValueWrapper func, in InterleaveDatasetParams()
89 FunctionDefHelper::AttrValueWrapper func_;
96 FunctionDefHelper::AttrValueWrapper MakeTensorSliceDatasetFunc( in MakeTensorSliceDatasetFunc()
99 return FunctionDefHelper::FunctionRef( in MakeTensorSliceDatasetFunc()
|
| D | dataset_utils_test.cc | 163 return FunctionDefHelper::Create( in TEST()
173 return FunctionDefHelper::Create( in TEST()
211 *fdef_base.add_function() = FunctionDefHelper::Create( in TEST()
220 *fdef_to_add.add_function() = FunctionDefHelper::Create( in TEST()
248 *flib.add_function() = FunctionDefHelper::Create( in TEST()
|
| D | parallel_interleave_dataset_op_test.cc | 30 FunctionDefHelper::AttrValueWrapper func, in ParallelInterleaveDatasetParams()
110 FunctionDefHelper::AttrValueWrapper func_;
118 FunctionDefHelper::AttrValueWrapper MakeTensorSliceDatasetFunc( in MakeTensorSliceDatasetFunc()
121 return FunctionDefHelper::FunctionRef( in MakeTensorSliceDatasetFunc()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer_test.cc | 92 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 …]
|
| D | meta_optimizer_test.cc | 263 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
269 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
276 FunctionDef quadratic_func = FunctionDefHelper::Create( in TEST_F()
412 FunctionDef my_mul = FunctionDefHelper::Create( in TEST_F()
421 FunctionDef my_fwd = FunctionDefHelper::Create( in TEST_F()
464 FunctionDef expected_my_mul = FunctionDefHelper::Create( in TEST_F()
471 FunctionDef expected_my_fwd = FunctionDefHelper::Create( in TEST_F()
514 FunctionDef my_func = FunctionDefHelper::Create( in TEST_F()
604 using FDH = FunctionDefHelper; in TEST_F()
624 FunctionDef mul_func_1 = FunctionDefHelper::Create( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util_test.cc | 151 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
192 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
202 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
272 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
279 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
373 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
380 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
427 FunctionDef identity_func = FunctionDefHelper::Create( in TEST_F()
455 FunctionDef sort_func = FunctionDefHelper::Create( in TEST_F()
486 FunctionDef identity_func = FunctionDefHelper::Create( in TEST_F()
[all …]
|
| D | build_xla_ops_pass_test.cc | 124 FunctionDef func = FunctionDefHelper::Create( in CreateFunctionDefLibWithConstFunction()
127 {}, /*node_def=*/{FunctionDefHelper::Const("one", 1.0f)}, in CreateFunctionDefLibWithConstFunction()
264 FunctionDef func = FunctionDefHelper::Create( in CreateFunctionDefLibWithInt32Input()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_testlib.h | 37 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>>
|
| D | graph_to_functiondef_test.cc | 75 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
112 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
152 FunctionDefHelper::Create("test_fn", // function name in TEST()
|
| D | op_def_builder.h | 31 class FunctionDefHelper; variable
159 friend class FunctionDefHelper;
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op_test.cc | 29 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()
|
| D | parallel_interleave_dataset_op_test.cc | 32 FunctionDefHelper::AttrValueWrapper func, in ParallelInterleaveDatasetParams()
107 FunctionDefHelper::AttrValueWrapper func_;
114 FunctionDefHelper::AttrValueWrapper MakeTensorSliceDatasetFunc( in MakeTensorSliceDatasetFunc()
117 return FunctionDefHelper::FunctionRef( in MakeTensorSliceDatasetFunc()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_grad.cc | 22 typedef FunctionDefHelper FDH;
|