| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 42 class AttrSlice; variable
69 std::string SummarizeAttrsHelper(AttrSlice attrs, StringPiece device);
140 class AttrSlice {
142 AttrSlice(const NodeDef& node_def); // NOLINT(runtime/explicit)
144 AttrSlice(); // Empty
145 explicit AttrSlice(const AttrValueMap* a);
174 bool EqualAttrs(AttrSlice other, Scratch* scratch) const;
198 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
200 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
202 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
[all …]
|
| D | node_def_util.cc | 48 AttrSlice::AttrSlice() : ndef_(nullptr) { in AttrSlice() function in tensorflow::AttrSlice
53 AttrSlice::AttrSlice(const NodeDef& node_def) in AttrSlice() function in tensorflow::AttrSlice
56 AttrSlice::AttrSlice(const AttrValueMap* a) : ndef_(nullptr), attrs_(a) {} in AttrSlice() function in tensorflow::AttrSlice
58 string SummarizeAttrsHelper(AttrSlice attrs, StringPiece device) { in SummarizeAttrsHelper()
85 string AttrSlice::SummarizeNode() const { in SummarizeNode()
91 string AttrSlice::DebugString() const { in DebugString()
144 const AttrValue* AttrSlice::Find(StringPiece attr_name) const { in Find()
165 const AttrValue* AttrSlice::FindByString(const string& attr_name) const { in FindByString()
174 Status AttrSlice::Find(StringPiece attr_name, in Find()
190 bool AttrSlice::EqualAttrs(AttrSlice other, Scratch* scratch) const { in EqualAttrs()
[all …]
|
| D | function_test.cc | 47 operator AttrSlice() { return AttrSlice(&map_); } // NOLINT(runtime/explicit) in operator AttrSlice()
236 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
273 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
379 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
459 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
531 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
546 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
556 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
609 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
620 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
[all …]
|
| D | function.h | 242 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
741 virtual Status Instantiate(const std::string& function_name, AttrSlice attrs,
744 Status Instantiate(const std::string& function_name, AttrSlice attrs, in Instantiate()
909 AttrSlice attrs);
926 const std::string& funcname, AttrSlice attrs,
928 std::string Canonicalize(const std::string& funcname, AttrSlice attrs);
973 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
1023 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,
1087 typedef std::function<Status(const AttrSlice& attrs, FunctionDef*)> Creator;
|
| D | function_testlib.h | 51 operator AttrSlice() { return AttrSlice(&map_); } // NOLINT(runtime/explicit) in AttrSlice() function
|
| D | kernel_def_util.h | 26 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs,
|
| D | function_handle_cache.h | 34 Status Instantiate(const string& function_name, AttrSlice attrs,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad.cc | 46 Status AbsGrad(const AttrSlice& attrs, FunctionDef* g) { in AbsGrad()
56 Status NegGrad(const AttrSlice& attrs, FunctionDef* g) { in NegGrad()
65 Status InvGrad(const AttrSlice& attrs, FunctionDef* g) { in InvGrad()
78 Status SquareGrad(const AttrSlice& attrs, FunctionDef* g) { in SquareGrad()
90 Status SqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in SqrtGrad()
104 Status RsqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in RsqrtGrad()
119 Status ExpGrad(const AttrSlice& attrs, FunctionDef* g) { in ExpGrad()
129 Status Expm1Grad(const AttrSlice& attrs, FunctionDef* g) { in Expm1Grad()
139 Status LogGrad(const AttrSlice& attrs, FunctionDef* g) { in LogGrad()
149 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 SplitVGrad(const AttrSlice& attrs, FunctionDef* g) { in SplitVGrad()
[all …]
|
| D | nn_grad.cc | 25 Status SoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in SoftmaxGrad()
50 Status LogSoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in LogSoftmaxGrad()
75 Status ReluGrad(const AttrSlice& attrs, FunctionDef* g) { in ReluGrad()
93 Status Relu6Grad(const AttrSlice& attrs, FunctionDef* g) { in Relu6Grad()
111 Status CrossEntropyGrad(const AttrSlice& attrs, FunctionDef* g) { in CrossEntropyGrad()
140 Status Conv2DGrad(const AttrSlice& attrs, FunctionDef* g) { in Conv2DGrad()
176 Status MaxPoolGrad(const AttrSlice& attrs, FunctionDef* g) { in MaxPoolGrad()
207 Status AvgPoolGrad(const AttrSlice& attrs, FunctionDef* g) { in AvgPoolGrad()
233 Status MaxPoolGradGrad(const AttrSlice& attrs, FunctionDef* g) { in MaxPoolGradGrad()
264 Status BiasAddGrad(const AttrSlice& attrs, FunctionDef* g) { in BiasAddGrad()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 78 const AttrSlice& func_attr() const;
88 friend Status MakeGrapplerFunctionItem(const FunctionDef&, const AttrSlice&,
98 AttrSlice func_attr,
107 AttrSlice func_attr_; // Attributes specific to function definition that
135 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
142 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
162 const AttrSlice& func_instantiation_attr,
|
| D | functions_test.cc | 91 func, AttrSlice(&func_instantiation_attr), &type_parameters)); in TEST_F()
102 func, AttrSlice(&func_instantiation_attr), &body_parameters)); in TEST_F()
132 AttrSlice(&func_instantiation_attr), in TEST_F()
213 AttrSlice(&func_instantiation_attr), in TEST_F()
308 AttrSlice(&func_instantiation_attr), in TEST_F()
373 AttrSlice(&func_instantiation_attr), in TEST_F()
425 AttrSlice(&func_instantiation_attr), in TEST_F()
464 AttrSlice(&func_instantiation_attr), in TEST_F()
491 AttrSlice(&func_instantiation_attr), in TEST_F()
540 AttrSlice(&func_instantiation_attr), in TEST_F()
[all …]
|
| D | functions.cc | 40 string func_name, string description, AttrSlice func_attr, in GrapplerFunctionItem()
111 const AttrSlice& GrapplerFunctionItem::func_attr() const { return func_attr_; } in func_attr()
157 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationTypeParameters()
193 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationBodyParameters()
222 const AttrSlice& func_instantiation_attr, in MakeGrapplerFunctionItem()
296 /*func_attr=*/AttrSlice(&func.attr()), std::move(arg_attr), in MakeGrapplerFunctionItem()
306 return MakeGrapplerFunctionItem(func, AttrSlice(), flib, graph_def_version, in MakeGrapplerFunctionItem()
336 auto attrs = AttrSlice(node); in ReplaceInputWithConst()
386 auto attrs = AttrSlice(node); in RemoveFunctionOutputs()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_def_utils.h | 26 class AttrSlice; variable
34 Status FunctionDefToBodyHelper(const FunctionDef& fdef, const AttrSlice& attrs,
42 const FunctionDef& fdef, const AttrSlice& attrs,
|
| D | function_def_utils.cc | 30 const FunctionDef& fdef, const AttrSlice& attrs, in FunctionDefToBodyHelper()
76 Status FunctionDefToBodyHelper(const FunctionDef& fdef, const AttrSlice& attrs, in FunctionDefToBodyHelper()
|
| D | process_function_library_runtime.h | 163 Status Instantiate(const string& function_name, AttrSlice attrs,
366 const string& function_name, AttrSlice attrs,
371 const string& function_name, AttrSlice attrs,
424 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 89 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in IsIndirectFunctionCall()
94 AttrSlice FunctionInstantiationAttributes(const FunctionDef& func, in FunctionInstantiationAttributes()
97 return AttrSlice(func_node); in FunctionInstantiationAttributes()
100 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in FunctionInstantiationAttributes()
101 return AttrSlice(&func_attr->func().attr()); in FunctionInstantiationAttributes()
106 return AttrSlice(); in FunctionInstantiationAttributes()
149 const auto attr = AttrSlice(&fdef.attr()); in MarkedNoSpecialize()
391 const AttrValue* func_attr = AttrSlice(node).Find("f"); in FindFunctionCall()
538 const AttrValue* tin = AttrSlice(func_node).Find("Tin"); in RemovePushedDownConstInputTypes()
562 const AttrValue* tout = AttrSlice(func_node).Find("Tout"); in RemoveUnusedOutputsTypes()
[all …]
|
| D | function_optimizer_test.cc | 1516 auto& func = AttrSlice(node).Find("f")->func(); in TEST_F()
1520 auto& tin = AttrSlice(node).Find("Tin")->list(); in TEST_F()
1521 auto& tout = AttrSlice(node).Find("Tout")->list(); in TEST_F()
1660 auto& func = AttrSlice(node).Find("f")->func(); in TEST_F()
1663 auto& tin = AttrSlice(node).Find("Tin")->list(); in TEST_F()
1664 auto& tout = AttrSlice(node).Find("Tout")->list(); in TEST_F()
2003 auto& func = AttrSlice(node).Find("f")->func(); in TEST_F()
2004 auto& tout = AttrSlice(node).Find("Tout")->list(); in TEST_F()
2010 auto& func = AttrSlice(node).Find("f")->func(); in TEST_F()
2011 auto& tout = AttrSlice(node).Find("Tout")->list(); in TEST_F()
[all …]
|
| D | scoped_allocator_optimizer.cc | 99 AttrSlice n_attrs = AttrSlice(*n); in CheckTypesAndGetShapes()
388 AttrSlice n_attrs = AttrSlice(*nd.from_node_def); in CheckExistingScopedAllocator()
674 AttrSlice first_slice(*ops[0]); in BuildReplacementOp()
1163 AttrSlice a_attrs = AttrSlice(*a); in operator ()()
1164 AttrSlice b_attrs = AttrSlice(*b); in operator ()()
1182 AttrSlice attrs = AttrSlice(n); in IsCollectiveNode()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 118 TF_CHECK_OK(GetNodeAttr(AttrSlice(&call_node_def.attr()), in TEST()
223 TF_CHECK_OK(GetNodeAttr(AttrSlice(&call_node_def.attr()), "shapes", &shapes)); in TEST()
306 AttrSlice(), &fld, &xla_fbody)); in TEST_F()
348 *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, &host_fbody)); in TEST_F()
504 AttrSlice(&host_func_attrs), &fld, in TEST_F()
550 AttrSlice(), &fld, &xla_fbody)); in TEST_F()
654 AttrSlice(&host_func_attrs), &fld, in TEST_F()
790 AttrSlice(&host_func_attrs), &fld, in TEST_F()
801 AttrSlice(&host_func_attrs), &fld, in TEST_F()
821 AttrSlice(), &fld, &xla_fbody)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.h | 41 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
63 const OpDef& sig, AttrSlice attrs,
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | optimizer_cse.cc | 66 AttrSlice::Scratch* scratch);
240 AttrSlice::Scratch* scratch) { in Equivalent()
295 AttrSlice::Scratch scratch; in Optimize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 136 AttrSlice attrs, FunctionLibraryRuntime::InstantiateOptions* opts) { in FillOutputDevices()
226 FillOutputDevices(*lib, *cpu_device, AttrSlice(&func_->attr()), &opts)); in Instantiate()
229 lib->Instantiate(func_->name(), AttrSlice(&func_->attr()), opts, handle)); in Instantiate()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | frontend_attributes_util.h | 34 GetFrontendAttributesFromAttrSlice(const AttrSlice& attrs);
|
| D | frontend_attributes_util.cc | 26 GetFrontendAttributesFromAttrSlice(const AttrSlice& attrs) { in GetFrontendAttributesFromAttrSlice()
|