/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 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()
|
D | functional_grad.cc | 24 Status MapAccumulateGrad(const AttrSlice& attrs, FunctionDef* ret) { in MapAccumulateGrad()
|
/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util.h | 126 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 …]
|
D | node_def_util.cc | 43 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 …]
|
D | function_test.cc | 44 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 …]
|
D | function.h | 229 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;
|
D | function_testlib.h | 43 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.cc | 54 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 …]
|
D | function_handle_cache.h | 34 Status Instantiate(const string& function_name, AttrSlice attrs,
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | functions.h | 140 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,
|
D | functions_test.cc | 84 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 …]
|
D | functions.cc | 61 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/ |
D | scoped_allocator_optimizer.cc | 84 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()
|
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 | function_optimizer.cc | 112 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/ |
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() 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/ |
D | optimizer_cse.cc | 63 AttrSlice::Scratch* scratch); 133 AttrSlice::Scratch* scratch) { in Equivalent() 188 AttrSlice::Scratch scratch; in Optimize()
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | cluster_function_library_runtime.h | 38 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, 49 const OpDef& sig, AttrSlice attrs,
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | process_function_library_runtime.h | 120 Status Instantiate(const string& function_name, AttrSlice attrs, 230 const string& function_name, AttrSlice attrs, 274 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
|
D | function.cc | 174 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/ |
D | partitioned_function_ops.cc | 130 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/ |
D | const_analysis.cc | 145 cond_function.name(), AttrSlice(&cond_function.attr()), &cond_handle)); in GetCompileTimeConstInputs() 147 body_function.name(), AttrSlice(&body_function.attr()), &body_handle)); in GetCompileTimeConstInputs()
|