| /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 | 32 {{"predicate", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeFilterNode()
47 {{"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeMapAndBatchNode()
57 {{"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeMapNode()
74 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelInterleaveNode()
89 {"f", FunctionDefHelper::FunctionRef(string(function_name))}, in MakeParallelMapNode()
|
| D | map_vectorization_test.cc | 60 *map_fn = FunctionDefHelper::Create( in AddMapFn()
81 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
93 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
300 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapAndBatchNode()
409 std::vector<std::pair<string, FunctionDefHelper::AttrValueWrapper>> attrs; in AddArbitraryInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer_test.cc | 127 FunctionDef my_identity_n = FunctionDefHelper::Create( in TEST_F()
202 FunctionDef x_times_two = FunctionDefHelper::Define( in TEST_F()
299 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
383 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
437 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
474 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
480 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
565 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
569 FunctionDefHelper::Const("zero", 0), in TEST_F()
570 FunctionDefHelper::Const("one", 1), in TEST_F()
[all …]
|
| D | meta_optimizer_test.cc | 258 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
264 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
271 FunctionDef quadratic_func = FunctionDefHelper::Create( in TEST_F()
414 FunctionDef my_mul = FunctionDefHelper::Create( in TEST_F()
423 FunctionDef my_fwd = FunctionDefHelper::Create( in TEST_F()
466 FunctionDef expected_my_mul = FunctionDefHelper::Create( in TEST_F()
473 FunctionDef expected_my_fwd = FunctionDefHelper::Create( in TEST_F()
516 FunctionDef my_func = FunctionDefHelper::Create( in TEST_F()
606 using FDH = FunctionDefHelper; in TEST_F()
627 FunctionDefHelper::Create("MyMul1", {"x:float", "y:float"}, {"z:float"}, in TEST_F()
[all …]
|
| D | implementation_selector_test.cc | 137 using FDH = FunctionDefHelper; in TEST_F()
|
| /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()
226 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
294 std::vector<FunctionDefHelper::Node> nodes = { in TEST_F()
312 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
382 TF_ASSERT_OK(flib.AddFunctionDef(FunctionDefHelper::Define( in TEST_F()
395 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
464 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
514 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils_test.cc | 85 return FunctionDefHelper::Create( in TEST()
95 return FunctionDefHelper::Create( in TEST()
133 *fdef_base.add_function() = FunctionDefHelper::Create( in TEST()
142 *fdef_to_add.add_function() = FunctionDefHelper::Create( in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_testlib.h | 37 std::pair<string, FunctionDefHelper::AttrValueWrapper>>& attrs) { in Attrs()
52 gtl::ArraySlice<std::pair<string, FunctionDefHelper::AttrValueWrapper>>
|
| D | op_def_builder.h | 31 class FunctionDefHelper; variable
155 friend class FunctionDefHelper;
|
| D | graph_to_functiondef_test.cc | 75 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
112 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
|
| D | function.h | 77 class FunctionDefHelper {
191 inline FunctionDefHelper::AttrValueWrapper::AttrValueWrapper(const char* val) { in AttrValueWrapper()
196 inline FunctionDefHelper::AttrValueWrapper::AttrValueWrapper( in AttrValueWrapper()
202 inline FunctionDefHelper::AttrValueWrapper::AttrValueWrapper(StringPiece val) { in AttrValueWrapper()
|
| D | graph_def_util_test.cc | 193 FunctionDefHelper::Create( in TEST()
212 FunctionDefHelper::Create( in TEST()
|
| D | function.cc | 1497 void FunctionDefHelper::AttrValueWrapper::InitFromString(StringPiece val) { in InitFromString()
1505 FunctionDefHelper::AttrValueWrapper FunctionDefHelper::FunctionRef( in FunctionRef()
1516 NodeDef FunctionDefHelper::Node::ToNodeDef() const { in ToNodeDef()
1536 FunctionDef FunctionDefHelper::Create( in Create()
1586 FunctionDef FunctionDefHelper::Create( in Create()
1596 FunctionDef FunctionDefHelper::Define(const string& name, in Define()
1665 FunctionDef FunctionDefHelper::Define(gtl::ArraySlice<string> arg_def, in Define()
|
| D | function_testlib.cc | 29 typedef FunctionDefHelper FDH;
|
| D | function_test.cc | 37 std::pair<string, FunctionDefHelper::AttrValueWrapper>> in Attrs()
50 typedef FunctionDefHelper FDH;
1353 using FDH = ::tensorflow::FunctionDefHelper; in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 614 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
669 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
675 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
925 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
1083 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
1230 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
1262 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
1390 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
1417 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
1526 *library_expected.add_function() = FunctionDefHelper::Create( in TEST()
[all …]
|
| D | mark_for_compilation_pass_test.cc | 199 FunctionDef compilable = FunctionDefHelper::Define( in TEST()
203 FunctionDefHelper::Define("UncompilableFn", {"n_a:float"}, {"n_c:float"}, in TEST()
266 std::vector<FunctionDefHelper::Node> nodes) { in GradForUnaryCwise()
272 *g = FunctionDefHelper::Define( in GradForUnaryCwise()
290 FunctionDefHelper::Const("one", 1.0f), in SupportedGrad()
304 FunctionDefHelper::Const("one", 1.0f), in UnsupportedGrad()
815 FunctionDef func = FunctionDefHelper::Create( in TEST()
820 {FunctionDefHelper::Const("shape_shape", 2), in TEST()
821 FunctionDefHelper::Const("minval", 1), in TEST()
822 FunctionDefHelper::Const("maxval", 20), in TEST()
|
| D | build_xla_ops_pass_test.cc | 123 FunctionDef func = FunctionDefHelper::Create( in CreateFunctionDefLibWithConstFunction()
126 {}, /*node_def=*/{FunctionDefHelper::Const("one", 1.0f)}, in CreateFunctionDefLibWithConstFunction()
|
| D | create_xla_launch_op_test.cc | 40 return FunctionDefHelper::Define( in XTimesY()
|
| D | resource_operation_safety_analysis_test.cc | 244 FunctionDef func = FunctionDefHelper::Create( in CreateFunctionDefLibWithConstFunction()
247 {}, /*node_def=*/{FunctionDefHelper::Const("one", 1.0f)}, in CreateFunctionDefLibWithConstFunction()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_grad.cc | 22 typedef FunctionDefHelper FDH;
|
| D | nn_grad.cc | 23 typedef FunctionDefHelper FDH;
|
| D | array_grad.cc | 22 typedef FunctionDefHelper FDH;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_testlib.cc | 27 typedef FunctionDefHelper FDH;
|