| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad.cc | 43 Status AbsGrad(const AttrSlice& attrs, FunctionDef* g) { in AbsGrad()
53 Status NegGrad(const AttrSlice& attrs, FunctionDef* g) { in NegGrad()
62 Status InvGrad(const AttrSlice& attrs, FunctionDef* g) { in InvGrad()
75 Status SquareGrad(const AttrSlice& attrs, FunctionDef* g) { in SquareGrad()
87 Status SqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in SqrtGrad()
101 Status RsqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in RsqrtGrad()
116 Status ExpGrad(const AttrSlice& attrs, FunctionDef* g) { in ExpGrad()
126 Status Expm1Grad(const AttrSlice& attrs, FunctionDef* g) { in Expm1Grad()
136 Status LogGrad(const AttrSlice& attrs, FunctionDef* g) { in LogGrad()
146 Status Log1pGrad(const AttrSlice& attrs, FunctionDef* g) { in Log1pGrad()
[all …]
|
| D | array_grad.cc | 33 Status ReshapeGrad(const AttrSlice& attrs, FunctionDef* g) { in ReshapeGrad()
54 Status SqueezeGrad(const AttrSlice& attrs, FunctionDef* g) { in SqueezeGrad()
73 Status IdentityGrad(const AttrSlice& attrs, FunctionDef* g) { in IdentityGrad()
92 Status PackGrad(const AttrSlice& attrs, FunctionDef* g) { in PackGrad()
118 Status UnpackGrad(const AttrSlice& attrs, FunctionDef* g) { in UnpackGrad()
142 Status ConcatGradHelper(const AttrSlice& attrs, FunctionDef* g, in ConcatGradHelper()
216 Status ConcatGrad(const AttrSlice& attrs, FunctionDef* g) { in ConcatGrad()
220 Status ConcatGradV2(const AttrSlice& attrs, FunctionDef* g) { in ConcatGradV2()
227 Status SplitGrad(const AttrSlice& attrs, FunctionDef* g) { in SplitGrad()
247 Status ArrayToListGrad(const AttrSlice& attrs, FunctionDef* g) { in ArrayToListGrad()
[all …]
|
| D | nn_grad.cc | 25 Status SoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in SoftmaxGrad()
52 Status ReluGrad(const AttrSlice& attrs, FunctionDef* g) { in ReluGrad()
70 Status Relu6Grad(const AttrSlice& attrs, FunctionDef* g) { in Relu6Grad()
88 Status CrossEntropyGrad(const AttrSlice& attrs, FunctionDef* g) { in CrossEntropyGrad()
117 Status Conv2DGrad(const AttrSlice& attrs, FunctionDef* g) { in Conv2DGrad()
153 Status MaxPoolGrad(const AttrSlice& attrs, FunctionDef* g) { in MaxPoolGrad()
184 Status AvgPoolGrad(const AttrSlice& attrs, FunctionDef* g) { in AvgPoolGrad()
210 Status MaxPoolGradGrad(const AttrSlice& attrs, FunctionDef* g) { in MaxPoolGradGrad()
241 Status BiasAddGrad(const AttrSlice& attrs, FunctionDef* g) { in BiasAddGrad()
|
| D | functional_grad.cc | 24 Status MapAccumulateGrad(const AttrSlice& attrs, FunctionDef* ret) { in MapAccumulateGrad()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 113 class AttrSlice {
115 AttrSlice(const NodeDef& node_def); // NOLINT(runtime/explicit)
117 AttrSlice(); // Empty
118 explicit AttrSlice(const AttrValueMap* a);
146 bool EqualAttrs(AttrSlice other, Scratch* scratch) const;
168 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
170 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
172 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
174 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
176 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
[all …]
|
| D | node_def_util.cc | 42 AttrSlice::AttrSlice() : ndef_(nullptr) { in AttrSlice() function in tensorflow::AttrSlice
47 AttrSlice::AttrSlice(const NodeDef& node_def) in AttrSlice() function in tensorflow::AttrSlice
50 AttrSlice::AttrSlice(const AttrValueMap* a) : ndef_(nullptr), attrs_(a) {} in AttrSlice() function in tensorflow::AttrSlice
52 static string SummarizeAttrsHelper(AttrSlice attrs, StringPiece device) { in SummarizeAttrsHelper()
79 string AttrSlice::SummarizeNode() const { in SummarizeNode()
103 const AttrValue* AttrSlice::Find(StringPiece attr_name) const { in Find()
124 Status AttrSlice::Find(StringPiece attr_name, in Find()
140 bool AttrSlice::EqualAttrs(AttrSlice other, Scratch* scratch) const { in EqualAttrs()
159 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
169 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
[all …]
|
| D | function_test.cc | 44 operator AttrSlice() { return AttrSlice(&map_); } // NOLINT(runtime/explicit) in operator AttrSlice()
191 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
228 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
334 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
414 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
486 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
546 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
557 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
568 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
578 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
[all …]
|
| D | function.h | 209 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
447 virtual Status Instantiate(const string& function_name, AttrSlice attrs,
450 Status Instantiate(const string& function_name, AttrSlice attrs, in Instantiate()
552 string Canonicalize(const string& funcname, AttrSlice attrs,
554 inline string Canonicalize(const string& funcname, AttrSlice attrs) { in Canonicalize()
572 AttrSlice attrs,
596 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,
659 typedef std::function<Status(const AttrSlice& attrs, FunctionDef*)> Creator;
|
| D | function_testlib.h | 43 operator AttrSlice() { return AttrSlice(&map_); } // NOLINT(runtime/explicit) in AttrSlice() function
|
| D | function.cc | 50 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def, in ArgNumType()
99 Status ValidateSignatureWithAttrs(const OpDef& sig, AttrSlice attr_values) { in ValidateSignatureWithAttrs()
152 AttrSlice attr_values) { in BuildInputArgIndex()
180 Status BuildNodeOutputIndex(const NodeDef& node, AttrSlice attrs, in BuildNodeOutputIndex()
210 Status InstantiateNode(const NodeDef& fnode, AttrSlice attrs) { in InstantiateNode()
303 const OpDef::ArgDef& ret_def, AttrSlice attrs, in AddReturnNode()
614 AttrSlice attr_slice(attrs); in AddDefaultAttrs()
627 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values, in InstantiateFunction()
672 s = helper.BuildNodeOutputIndex(fdef.node_def(i), AttrSlice(&node_attrs[i]), in InstantiateFunction()
681 s = helper.InstantiateNode(fdef.node_def(i), AttrSlice(&node_attrs[i])); in InstantiateFunction()
[all …]
|
| D | shape_inference.h | 329 AttrSlice attrs() const { return AttrSlice(*node_def_); } in attrs()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | optimizer_cse.cc | 62 AttrSlice::Scratch* scratch);
132 AttrSlice::Scratch* scratch) { in Equivalent()
187 AttrSlice::Scratch scratch; in Optimize()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.h | 36 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
47 const OpDef& sig, AttrSlice attrs,
|
| D | cluster_function_library_runtime.cc | 29 const OpDef& sig, AttrSlice attrs, in ConstructFunctionGraph()
122 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 153 Status Instantiate(const string& function_name, AttrSlice attrs,
228 Status FunctionDefToBody(const FunctionDef& fdef, AttrSlice attrs,
237 AttrValueMap FixAttrs(const AttrSlice& attrs);
382 Instantiate(ndef.op(), AttrSlice(&ndef.attr()), options, &handle)); in CreateKernel()
415 const FunctionDef& fdef, AttrSlice attrs, in FunctionDefToBody()
442 TF_RETURN_IF_ERROR(creator(AttrSlice(&func.attr()), &grad_fdef)); in InstantiateSymbolicGradient()
444 FunctionDefToBody(grad_fdef, AttrSlice(&func.attr()), lib_def, g_body)); in InstantiateSymbolicGradient()
453 Instantiate(func.name(), AttrSlice(&func.attr()), options, &f_handle)); in InstantiateSymbolicGradient()
473 const string& function_name, AttrSlice attrs, in Instantiate()
506 return Instantiate(grad, AttrSlice(&func.attr()), options, handle); in Instantiate()
[all …]
|
| D | function.h | 165 const FunctionDef& fdef, const AttrSlice& attrs,
|
| D | parallel_concat_optimizer.cc | 52 AttrSlice n_attrs = n->attrs(); in Run()
|
| D | process_function_library_runtime.h | 108 Status Instantiate(const string& function_name, AttrSlice attrs,
|
| D | accumulate_n_optimizer.cc | 76 AttrSlice n_attrs = n->attrs(); in rewriteNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | test_util.cc | 34 InstantiateFunction(*fdef, AttrSlice(), get_func_sig, &inst)); in InstantiateFunctionForTest()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 241 ctx, lib->Instantiate(kGradientOp, AttrSlice(def()), &handle), done); in ComputeAsync()
310 lib->Instantiate(func_.name(), AttrSlice(&attr_values), in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 55 flr->Instantiate(ndef.op(), AttrSlice(&ndef.attr()), &handle)); in CreateXlaLaunchOp()
|
| D | mark_for_compilation_pass.cc | 119 lib_runtime->Instantiate(call_def.op(), AttrSlice(call_def), &handle); in IsCompilableCall()
128 if (GetNodeAttr(AttrSlice(&fdef.attr()), "_noinline", &noinline).ok() && in IsCompilableCall()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | function_ops.cc | 72 ctx, lib->Instantiate(kGradientOp, AttrSlice(&def().attr()), &handle_), in ComputeAsync()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | runtime.h | 113 const AttrValue* found = AttrSlice(m).Find(attr_name); in SetInAttrValueMap()
|