Home
last modified time | relevance | path

Searched refs:AttrSlice (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/core/ops/
Dmath_grad.cc43 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 …]
Darray_grad.cc33 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 …]
Dnn_grad.cc25 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()
Dfunctional_grad.cc24 Status MapAccumulateGrad(const AttrSlice& attrs, FunctionDef* ret) { in MapAccumulateGrad()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h126 class AttrSlice {
128 AttrSlice(const NodeDef& node_def); // NOLINT(runtime/explicit)
130 AttrSlice(); // Empty
131 explicit AttrSlice(const AttrValueMap* a);
159 bool EqualAttrs(AttrSlice other, Scratch* scratch) const;
181 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
183 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
185 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
187 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
189 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
[all …]
Dnode_def_util.cc43 AttrSlice::AttrSlice() : ndef_(nullptr) { in AttrSlice() function in tensorflow::AttrSlice
48 AttrSlice::AttrSlice(const NodeDef& node_def) in AttrSlice() function in tensorflow::AttrSlice
51 AttrSlice::AttrSlice(const AttrValueMap* a) : ndef_(nullptr), attrs_(a) {} in AttrSlice() function in tensorflow::AttrSlice
53 static string SummarizeAttrsHelper(AttrSlice attrs, StringPiece device) { in SummarizeAttrsHelper()
80 string AttrSlice::SummarizeNode() const { in SummarizeNode()
155 const AttrValue* AttrSlice::Find(StringPiece attr_name) const { in Find()
176 Status AttrSlice::Find(StringPiece attr_name, in Find()
192 bool AttrSlice::EqualAttrs(AttrSlice other, Scratch* scratch) const { in EqualAttrs()
211 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
221 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
[all …]
Dfunction_test.cc44 operator AttrSlice() { return AttrSlice(&map_); } // NOLINT(runtime/explicit) in operator AttrSlice()
233 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
270 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
376 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
456 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
528 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
543 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
553 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result)); in TEST()
606 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
617 HasError(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result), in TEST()
[all …]
Dfunction.h229 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
586 virtual Status Instantiate(const string& function_name, AttrSlice attrs,
589 Status Instantiate(const string& function_name, AttrSlice attrs, in Instantiate()
700 AttrSlice attrs);
710 string Canonicalize(const string& funcname, AttrSlice attrs,
712 inline string Canonicalize(const string& funcname, AttrSlice attrs) { in Canonicalize()
730 AttrSlice attrs,
754 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,
818 typedef std::function<Status(const AttrSlice& attrs, FunctionDef*)> Creator;
Dfunction_testlib.h43 operator AttrSlice() { return AttrSlice(&map_); } // NOLINT(runtime/explicit) in AttrSlice() function
Dkernel_def_util.h26 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs,
Dfunction.cc54 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def, in ArgNumType()
103 Status ValidateSignatureWithAttrs(const OpDef& sig, AttrSlice attr_values) { in ValidateSignatureWithAttrs()
155 Status BuildInputArgIndex(const OpDef::ArgDef& arg_def, AttrSlice attr_values, in BuildInputArgIndex()
188 Status BuildNodeOutputIndex(const NodeDef& node, AttrSlice attrs, in BuildNodeOutputIndex()
218 Status InstantiateNode(const NodeDef& fnode, AttrSlice attrs) { in InstantiateNode()
312 const OpDef::ArgDef& ret_def, AttrSlice attrs, in AddReturnNode()
668 AttrSlice attr_slice(attrs); in AddDefaultAttrs()
682 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values, in InstantiateFunction()
731 s = helper.BuildNodeOutputIndex(fdef.node_def(i), AttrSlice(&node_attrs[i]), in InstantiateFunction()
741 s = helper.InstantiateNode(fdef.node_def(i), AttrSlice(&node_attrs[i])); in InstantiateFunction()
[all …]
Dfunction_handle_cache.h34 Status Instantiate(const string& function_name, AttrSlice attrs,
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h140 explicit GrapplerFunctionItemInstantiation(AttrSlice func_instantiation_attr) in GrapplerFunctionItemInstantiation()
152 const AttrSlice func_instantiation_attr_; // do not own
173 const AttrSlice& func_attr() const;
182 friend Status MakeGrapplerFunctionItem(const FunctionDef&, const AttrSlice&,
192 AttrSlice func_attr,
200 AttrSlice func_attr_; // Attributes specific to function definition that
225 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
232 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
261 const AttrSlice& func_instantiation_attr,
Dfunctions_test.cc84 func, AttrSlice(&func_instantiation_attr), &type_parameters)); in TEST_F()
93 func, AttrSlice(&func_instantiation_attr), &body_parameters)); in TEST_F()
248 AttrSlice(&func_instantiation_attr), in TEST_F()
330 AttrSlice(&func_instantiation_attr), in TEST_F()
421 AttrSlice(&func_instantiation_attr), in TEST_F()
484 AttrSlice(&func_instantiation_attr), in TEST_F()
533 AttrSlice(&func_instantiation_attr), in TEST_F()
598 AttrSlice(&func_instantiation_attr), in TEST_F()
637 AttrSlice(&func_instantiation_attr), in TEST_F()
664 AttrSlice(&func_instantiation_attr), in TEST_F()
[all …]
Dfunctions.cc61 const AttrSlice& func_instantiation_attr, NodeDef* node) { in ResolveFunctionBodyNodeAttrPlaceholders()
311 string func_name, string description, AttrSlice func_attr, in GrapplerFunctionItem()
383 const AttrSlice& GrapplerFunctionItem::func_attr() const { return func_attr_; } in func_attr()
424 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationTypeParameters()
452 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationBodyParameters()
482 const AttrSlice& func_instantiation_attr, in MakeGrapplerFunctionItem()
650 /*func_attr=*/AttrSlice(&func.attr()), std::move(inputs), in MakeGrapplerFunctionItem()
660 return MakeGrapplerFunctionItem(func, AttrSlice(), flib, graph_def_version, in MakeGrapplerFunctionItem()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc84 AttrSlice n_attrs = AttrSlice(*n); in CheckTypesAndGetShapes()
145 AttrSlice inode_attrs = AttrSlice(*inode); in GetInputs()
224 AttrSlice n_attrs = AttrSlice(*nd.from_node_def); in CheckExistingScopedAllocator()
423 AttrSlice first_slice(*ops[0]); in BuildReplacementOp()
892 AttrSlice a_attrs = AttrSlice(*a); in operator ()()
893 AttrSlice b_attrs = AttrSlice(*b); in operator ()()
911 AttrSlice attrs = AttrSlice(n); in IsCollectiveNode()
Dfunction_optimizer_test.cc1516 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 …]
Dfunction_optimizer.cc112 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); in IsIndirectFunctionCall()
117 AttrSlice FunctionInstantiationAttributes(const FunctionDef& func, in FunctionInstantiationAttributes()
120 return AttrSlice(func_node); in FunctionInstantiationAttributes()
123 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); in FunctionInstantiationAttributes()
124 return AttrSlice(&func_attr->func().attr()); in FunctionInstantiationAttributes()
129 return AttrSlice(); in FunctionInstantiationAttributes()
479 const AttrValue* func_attr = AttrSlice(node).Find("f"); in FindFunctionCall()
627 const AttrValue* tin = AttrSlice(func_node).Find("Tin"); in RemovePushedDownConstInputTypes()
652 const AttrValue* tout = AttrSlice(func_node).Find("Tout"); in RemoveUnusedOutputsTypes()
732 const AttrSlice& func_instantiation_attr, in InitializeFunctionSpecializationSignature()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc118 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()
305 *fld.Find("cluster_rewritten"), AttrSlice(), &fld, in TEST_F()
352 *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, in TEST_F()
437 *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, in TEST_F()
481 *fld.Find("cluster_rewritten"), AttrSlice(), &fld, in TEST_F()
498 *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, in TEST_F()
609 *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, in TEST_F()
659 *fld.Find("cluster_rewritten"), AttrSlice(), &fld, in TEST_F()
768 *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/graph/
Doptimizer_cse.cc63 AttrSlice::Scratch* scratch);
133 AttrSlice::Scratch* scratch) { in Equivalent()
188 AttrSlice::Scratch scratch; in Optimize()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.h38 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
49 const OpDef& sig, AttrSlice attrs,
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.h120 Status Instantiate(const string& function_name, AttrSlice attrs,
230 const string& function_name, AttrSlice attrs,
274 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
Dfunction.cc174 Status Instantiate(const string& function_name, AttrSlice attrs,
215 const string& function_name, AttrSlice attrs, in Instantiate()
317 Status Instantiate(const string& function_name, AttrSlice attrs,
403 Status FunctionDefToBody(const FunctionDef& fdef, AttrSlice attrs,
412 AttrValueMap FixAttrs(const AttrSlice& attrs);
569 Instantiate(ndef.op(), AttrSlice(&ndef.attr()), options, &handle)); in CreateKernel()
600 const FunctionDef& fdef, AttrSlice attrs, in FunctionDefToBody()
627 TF_RETURN_IF_ERROR(creator(AttrSlice(&func.attr()), &grad_fdef)); in InstantiateSymbolicGradient()
629 FunctionDefToBody(grad_fdef, AttrSlice(&func.attr()), lib_def, g_body)); in InstantiateSymbolicGradient()
638 Instantiate(func.name(), AttrSlice(&func.attr()), options, &f_handle)); in InstantiateSymbolicGradient()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc130 AttrSlice attrs, FunctionLibraryRuntime::InstantiateOptions* opts) { in FillOutputDevices()
203 FillOutputDevices(*lib, *cpu_device, AttrSlice(&func_->attr()), &opts)); in Instantiate()
206 lib->Instantiate(func_->name(), AttrSlice(&func_->attr()), opts, handle)); in Instantiate()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis.cc145 cond_function.name(), AttrSlice(&cond_function.attr()), &cond_handle)); in GetCompileTimeConstInputs()
147 body_function.name(), AttrSlice(&body_function.attr()), &body_handle)); in GetCompileTimeConstInputs()

1234