Home
last modified time | relevance | path

Searched refs:NameAttrList (Results 1 – 25 of 98) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.h118 static Status Create(const NameAttrList& func, OpKernelContext* ctx,
127 static Status Create(const NameAttrList& func, OpKernelContext* ctx,
137 static Status Create(const NameAttrList& func, OpKernelContext* ctx,
149 const NameAttrList& func() { return func_; } in func()
155 CapturedFunction(const NameAttrList& func,
159 const NameAttrList func_;
Dgenerator_dataset_op.h35 NameAttrList init_func_;
36 NameAttrList next_func_;
37 NameAttrList finalize_func_;
Dflat_map_dataset_op.cc53 const NameAttrList& func, in Dataset()
265 const NameAttrList func_;
274 NameAttrList func_;
Dfilter_dataset_op.cc97 const NameAttrList& func, in Dataset()
289 const NameAttrList func_;
295 NameAttrList func_;
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-name-attr-list.pbtxt1 path: "tensorflow.NameAttrList"
4 name: "NameAttrList"
16 type_name: ".tensorflow.NameAttrList.AttrEntry"
Dtensorflow.-attr-value.pbtxt70 type_name: ".tensorflow.NameAttrList"
144 type_name: ".tensorflow.NameAttrList"
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_window_dataset_op.cc68 const NameAttrList& key_func, const NameAttrList& reduce_func, in Dataset()
69 const NameAttrList& window_size_func, in Dataset()
516 const NameAttrList key_func_;
517 const NameAttrList reduce_func_;
518 const NameAttrList window_size_func_;
528 NameAttrList key_func_;
529 NameAttrList reduce_func_;
530 NameAttrList window_size_func_;
Dgroup_by_reducer_dataset_op.cc394 const NameAttrList& key_func() const { return captured_key_func_->func(); } in key_func()
396 const NameAttrList& init_func() const { in init_func()
400 const NameAttrList& reduce_func() const { in reduce_func()
404 const NameAttrList& finalize_func() const { in finalize_func()
441 NameAttrList key_func_;
442 NameAttrList init_func_;
443 NameAttrList reduce_func_;
444 NameAttrList finalize_func_;
Dtake_while_dataset_op.cc95 const NameAttrList& func, in Dataset()
237 const NameAttrList func_;
242 NameAttrList func_;
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.h107 const NameAttrList& function);
120 const NameAttrList function_;
154 const NameAttrList function_;
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.h59 NameAttrList cond_name_attr_;
60 NameAttrList body_name_attr_;
Dif_op.h50 NameAttrList then_branch_;
51 NameAttrList else_branch_;
/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.cc121 const NameAttrList& func = a.func(); in AttrValueHash()
148 const NameAttrList& af = a.func(); in AreAttrValuesEqual()
149 const NameAttrList& bf = b.func(); in AreAttrValuesEqual()
194 string SummarizeFunc(const NameAttrList& func) { in SummarizeFunc()
536 void SetAttrValue(const NameAttrList& value, AttrValue* out) { in SetAttrValue()
540 void SetAttrValue(gtl::ArraySlice<NameAttrList> value, AttrValue* out) { in SetAttrValue()
566 for (const NameAttrList& func : val.list().func()) { in HasPlaceHolder()
594 for (NameAttrList& func : *value->mutable_list()->mutable_func()) { in SubstitutePlaceholders()
Dattr_value_util.h35 class NameAttrList; variable
68 void SetAttrValue(const NameAttrList& value, AttrValue* out);
86 void SetAttrValue(gtl::ArraySlice<NameAttrList> value, AttrValue* out);
Dnode_def_util.h83 void AddNodeAttr(StringPiece name, const NameAttrList& value,
111 void AddNodeAttr(StringPiece name, gtl::ArraySlice<NameAttrList> value,
233 const NameAttrList** value); // type: "func"
237 NameAttrList* value); // type: "func"
239 std::vector<NameAttrList>* value); // type: "list(func)"
Dattr_value.proto26 repeated NameAttrList func = 9; // "list(attr)"
44 NameAttrList func = 10;
59 message NameAttrList { message
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.h72 const NameAttrList& function,
118 const NameAttrList& function,
124 const XlaCompiler::Options& options, const NameAttrList& function,
Dencapsulate_util.h76 const NameAttrList& func_name_attrs, Node* node, in XlaClusterInfo()
85 const NameAttrList func_name_attrs;
Dextract_outside_compilation_pass_test.cc117 NameAttrList shape_inference_graph; in TEST()
241 const string &xla_cluster_name, const NameAttrList &func_name_attrs, in ExtractOutsideCompilationTest()
294 NameAttrList name_attrs; in TEST_F()
334 NameAttrList shape_inference_graph; in TEST_F()
422 NameAttrList name_attrs; in TEST_F()
470 NameAttrList name_attrs; in TEST_F()
573 NameAttrList true_fn; in TEST_F()
575 NameAttrList false_fn; in TEST_F()
593 NameAttrList name_attrs; in TEST_F()
732 NameAttrList cond_fn; in TEST_F()
[all …]
Dxla_compilation_cache.cc118 const NameAttrList& function, in BuildSignature()
168 const XlaCompiler::Options& options, const NameAttrList& function, in Compile()
205 NameAttrList name; in CompileSingleOp()
227 const XlaCompiler::Options& options, const NameAttrList& function, in CompileImpl()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.h26 class NameAttrList; variable
58 std::unique_ptr<NameAttrList> func_;
Dfunctional_ops.cc38 Status Instantiate(FunctionLibraryRuntime* lib, const NameAttrList& func, in Instantiate()
161 NameAttrList then_func_;
162 NameAttrList else_func_;
252 std::vector<NameAttrList> branch_funcs_;
352 NameAttrList cond_func_;
353 NameAttrList body_func_;
503 const NameAttrList* func; in ForOp()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_while_test.cc77 NameAttrList then_func; in WhileWithIfBody()
79 NameAttrList else_func; in WhileWithIfBody()
196 NameAttrList cond_func; in IfWithWhileThen()
198 NameAttrList body_func; in IfWithWhileThen()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc44 NameAttrList* then_fn, NameAttrList* else_fn) { in FindIfThenAndElse()
48 const NameAttrList* result; in FindIfThenAndElse()
108 NameAttrList then_fn; in TEST()
109 NameAttrList else_fn; in TEST()
184 Status FindWhileCondAndBody(const GraphDef& graph, NameAttrList* cond, in FindWhileCondAndBody()
185 NameAttrList* body) { in FindWhileCondAndBody()
188 const NameAttrList* result; in FindWhileCondAndBody()
261 NameAttrList cond_fn, body_fn; in TEST()
413 NameAttrList cond_fn, body_fn; in TEST()
533 NameAttrList cond_fn, body_fn; in TEST()
[all …]
Dside_effect_util.cc45 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode()
55 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode()

1234