| /external/tensorflow/tensorflow/core/ops/ |
| nn_grad.cc | 23 typedef FunctionDefHelper FDH;
27 *g = FDH::Define(
40 FDH::Const<int32>("indices", {-1}),
52 *g = FDH::Define(
64 FDH::Const<int32>("indices", {-1}),
77 *g = FDH::Define(
95 *g = FDH::Define(
113 *g = FDH::Define(
127 FDH::Const("neg1", -1),
142 *g = FDH::Define
[all...] |
| functional_grad.cc | 22 typedef FunctionDefHelper FDH;
34 auto grad = FDH::FunctionRef("SymbolicGradient",
38 *ret = FDH::Define(
|
| array_grad.cc | 22 typedef FunctionDefHelper FDH;
35 *g = FDH::Define(
56 *g = FDH::Define(
75 *g = FDH::Define(
94 *g = FDH::Create(
120 *g = FDH::Define(
164 std::vector<FDH::Node> nodes{
183 *g = FDH::Create(
198 *g = FDH::Create(
229 *g = FDH::Define
[all...] |
| math_grad.cc | 22 typedef FunctionDefHelper FDH;
25 Status GradForUnaryCwise(FunctionDef* g, std::vector<FDH::Node> nodes) {
31 *g = FDH::Define(
78 FDH::Const("c", 2LL),
92 FDH::Const("const", 0.5f),
106 FDH::Const("const", -.5f),
149 FDH::Const("const", 1.0f),
183 FDH::Const("const", 1.0f),
218 FDH::Const("const", 1.0f),
232 FDH::Const("const", 1.0f)
[all...] |
| array_grad_test.cc | 28 using FDH = FunctionDefHelper;
45 {{"f", FDH::FunctionRef("Pack",
86 {{"f", FDH::FunctionRef("Unpack",
126 {{"f", FDH::FunctionRef("Concat", {{"N", 2}, {"T", T}})},
150 {{"f", FDH::FunctionRef("ConcatV2", {{"N", 2}, {"T", T}})},
222 {{"f", FDH::FunctionRef(
253 {{"f", FDH::FunctionRef("SplitV", {{"split_dim", dim},
315 {{"f", FDH::FunctionRef("Reshape", {{"T", T}})},
354 {{"f", FDH::FunctionRef("ExpandDims", {{"T", T}})},
391 {{"f", FDH::FunctionRef("Squeeze", {{"T", T}})}
[all...] |
| math_grad_test.cc | 30 using FDH = FunctionDefHelper;
42 Status Unary(const FDH::Node& op_node, const Tensor& x, const DataType dst,
50 auto test = FDH::Define("Test", {adef("x", src)}, {adef("l", dst)}, {},
53 FDH::Const("zero", 0),
54 FDH::Const("one", 1),
61 auto grad = FDH::Define(
64 FDH::Const("one", 1),
70 {"f", FDH::FunctionRef("Test")},
97 const FDH::Node op_node = {{"y"}, op, {"x"}, {{"T", x.dtype()}}};
110 const FDH::Node op_node =
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| function_testlib.cc | 29 typedef FunctionDefHelper FDH;
49 gtl::ArraySlice<std::pair<string, FDH::AttrValueWrapper>> attrs,
61 return FDH::Define(
78 return FDH::Define(
97 return FDH::Define(
121 return FDH::Define(
141 return FDH::Create(
181 return FDH::Create(
212 return FDH::Create(
230 return FDH::Create
[all...] |
| function_test.cc | 50 typedef FunctionDefHelper FDH;
67 auto fdef = FDH::Create(
114 auto fdef = FDH::Create(
160 auto fdef = FDH::Create(
209 auto fdef = FDH::Create(
246 auto fdef = FDH::Create(
300 auto fdef = FDH::Create(
314 {{"func", FDH::FunctionRef("Square", {{"T", "$T"}})},
347 auto fdef = FDH::Define(
416 auto fdef = FDH::Create
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| implementation_selector_test.cc | 137 using FDH = FunctionDefHelper;
141 FunctionDef boost_1 = FDH::Create(
144 FDH::Const("one", 1.0f)},
152 FunctionDef boost_1_gradient = FDH::Create(
154 {FDH::Const("two", 2.0f),
164 FunctionDef boost_2_func = FDH::Create(
166 {FDH::Const("four", 4.0f),
168 FDH::Const("one", 1.0f),
169 FDH::Const("two", 2.0f)},
177 FunctionDef boost_2_gradient = FDH::Create
[all...] |
| function_optimizer_test.cc | 712 using FDH = FunctionDefHelper;
731 {"f", FDH::FunctionRef("MyMul", {{"T", DT_FLOAT}})}},
772 using FDH = FunctionDefHelper;
816 {"f", FDH::FunctionRef("MyMul", {{"T", DT_FLOAT}})}},
823 {"f", FDH::FunctionRef("MyMul", {{"T", DT_FLOAT}})}},
[all...] |
| meta_optimizer_test.cc | 606 using FDH = FunctionDefHelper;
658 {{"f", FDH::FunctionRef("MyMul2", {})},
793 using FDH = FunctionDefHelper;
832 {{"f", FDH::FunctionRef("MyMul2", {})},
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function_testlib.cc | 27 typedef FunctionDefHelper FDH;
48 return FDH::Define(
101 return FDH::Define(
|
| function_test.cc | 57 using FDH = FunctionDefHelper;
442 using FDH = ::tensorflow::FunctionDefHelper;
458 {"f", FDH::FunctionRef("XTimesFour", {{"T", DT_FLOAT}})}}}},
496 FunctionDef stateful_func = FDH::Define(
506 {FDH::Const<int32>("shape", gtl::ArraySlice<int32>({1})),
507 FDH::Const<int32>("minval", 0),
508 FDH::Const<int32>("maxval", 10),
[all...] |
| lower_if_while_test.cc | 39 typedef FunctionDefHelper FDH;
58 return FDH::Define(
84 return FDH::Define(
202 return FDH::Define(
|
| lower_if_op_test.cc | 144 using FDH = ::tensorflow::FunctionDefHelper;
149 return FDH::Create(
|
| placer_test.cc | 50 using FDH = ::tensorflow::FunctionDefHelper;
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| mutable_graph_view_test.cc | 34 using FDH = FunctionDefHelper;
71 absl::Span<const std::pair<string, FDH::AttrValueWrapper>> attrs,
[all...] |
