| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.h | 144 std::pair<NameAttrList, std::vector<UncompilableNodeInfo>>>;
206 NameAttrList* encapsulating_function = nullptr,
211 NameAttrList* encapsulating_function = nullptr,
215 NameAttrList* encapsulating_function,
220 NameAttrList* encapsulating_function,
228 NameAttrList* encapsulating_function,
235 const std::string& call_name, NameAttrList* encapsulating_function,
274 NameAttrList* encapsulating_function,
293 const NameAttrList& function,
|
| D | compilability_check_util_test.cc | 138 uncompilable_nodes.find(NameAttrList().ShortDebugString()); in TEST_F()
173 NameAttrList function; in TEST_F()
216 NameAttrList compilable; in TEST_F()
218 NameAttrList uncompilable; in TEST_F()
245 NameAttrList function; in TEST_F()
286 NameAttrList uncompilable_fn1_attr; in TEST_F()
288 NameAttrList uncompilable_fn2_attr; in TEST_F()
322 NameAttrList function_one; in TEST_F()
346 NameAttrList function_two; in TEST_F()
438 NameAttrList b_name_attr; in TEST_F()
[all …]
|
| D | compilability_check_util.cc | 80 const NameAttrList* name_attr; in MakeCallNodeFromAttribute()
90 std::vector<NameAttrList> attr_lists; in MakeCallNodesFromAttribute()
233 NameAttrList* encapsulating_function, in IsCompilableIf()
252 NameAttrList* encapsulating_function, in IsCompilableCase()
279 NameAttrList* encapsulating_function, in IsCompilableWhile()
298 const std::string& call_name, NameAttrList* encapsulating_function, in ExtractNodeDefAndCheckCompilability()
329 NameAttrList* encapsulating_function, in IsCompilableCall()
343 NameAttrList function; in IsCompilableCall()
400 NameAttrList* encapsulating_function, in IsCompilableNode()
579 NameAttrList* encapsulating_function, in MaybeMarkUncompilableNode()
[all …]
|
| D | xla_compilation_cache.h | 73 const NameAttrList& function,
122 const NameAttrList& function,
128 const XlaCompiler::Options& options, const NameAttrList& function,
|
| D | encapsulate_util.h | 100 const NameAttrList& func_name_attrs, Node* node, in XlaClusterInfo()
109 const NameAttrList func_name_attrs;
|
| D | xla_compilation_cache_test.cc | 28 NameAttrList fn; in TEST()
58 NameAttrList fn; in TEST()
84 NameAttrList fn; in BM_BuildSignature()
|
| D | rearrange_function_argument_pass_test.cc | 94 NameAttrList f; in TEST()
101 NameAttrList cond_fn, body_fn; in TEST()
114 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST()
214 NameAttrList cond_fn, body_fn; in TEST()
225 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST()
|
| D | extract_outside_compilation_pass_test.cc | 117 NameAttrList shape_inference_graph; in TEST()
241 const string &xla_cluster_name, const NameAttrList &func_name_attrs, in ExtractOutsideCompilationTest()
295 NameAttrList name_attrs; in TEST_F()
330 NameAttrList shape_inference_graph; in TEST_F()
413 NameAttrList name_attrs; in TEST_F()
468 NameAttrList true_fn; in TEST_F()
470 NameAttrList false_fn; in TEST_F()
488 NameAttrList name_attrs; in TEST_F()
618 NameAttrList cond_fn; in TEST_F()
620 NameAttrList body_fn; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-name-attr-list.pbtxt | 1 path: "tensorflow.NameAttrList"
4 name: "NameAttrList"
16 type_name: ".tensorflow.NameAttrList.AttrEntry"
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.h | 52 const NameAttrList& function, bool has_ref_vars);
65 const NameAttrList function_;
101 const NameAttrList function_;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.h | 87 NameAttrList&& func, Params params,
91 const NameAttrList& func() const { return func_; } in func()
114 FunctionMetadata(NameAttrList&& func, Params params) in FunctionMetadata()
119 NameAttrList func_;
168 const NameAttrList& func() const { return metadata_->func(); } in func()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.h | 57 NameAttrList cond_name_attr_;
58 NameAttrList body_name_attr_;
|
| D | if_op.h | 50 NameAttrList then_branch_;
51 NameAttrList else_branch_;
|
| D | case_op.h | 59 std::pair<std::vector<NameAttrList>, xla::XlaOp> GetPrunedBranchesAndIndex(
62 std::vector<NameAttrList> unpruned_branches_;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_utils.h | 30 class NameAttrList; variable
53 NameAttrList* function);
|
| D | lower_while_op.cc | 65 static Status Run(Node* while_op, const NameAttrList& cond_fn, in Run()
66 const NameAttrList& body_fn, int parallel_iterations, in Run()
77 LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn,
78 const NameAttrList& body_fn, int parallel_iterations,
177 LowerWhileHelper::LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn, in LowerWhileHelper()
178 const NameAttrList& body_fn, in LowerWhileHelper()
|
| D | lower_if_op.cc | 42 CondBuilder(Node* if_op, const NameAttrList& then_fn,
43 const NameAttrList& else_fn, bool keep_node_fetchable,
113 CondBuilder::CondBuilder(Node* if_op, const NameAttrList& then_fn, in CondBuilder()
114 const NameAttrList& else_fn, bool keep_node_fetchable, in CondBuilder()
|
| D | lower_functional_ops_test.cc | 89 NameAttrList then_func; in WhileWithIfBody()
91 NameAttrList else_func; in WhileWithIfBody()
208 NameAttrList cond_func; in IfWithWhileThen()
210 NameAttrList body_func; in IfWithWhileThen()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 47 NameAttrList* then_fn, NameAttrList* else_fn) { in FindIfThenAndElse()
51 const NameAttrList* result; in FindIfThenAndElse()
143 NameAttrList then_fn; in CheckGraphDef()
144 NameAttrList else_fn; in CheckGraphDef()
292 Status FindWhileCondAndBody(const GraphDef& graph, NameAttrList* cond, in FindWhileCondAndBody()
293 NameAttrList* body) { in FindWhileCondAndBody()
296 const NameAttrList* result; in FindWhileCondAndBody()
369 NameAttrList cond_fn, body_fn; in TEST()
519 NameAttrList cond_fn, body_fn; in TEST()
637 NameAttrList cond_fn, body_fn; in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.h | 26 class NameAttrList; variable
58 std::unique_ptr<NameAttrList> func_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 46 class NameAttrList; variable
97 void AddNodeAttr(StringPiece name, const NameAttrList& value,
125 void AddNodeAttr(StringPiece name, gtl::ArraySlice<NameAttrList> value,
256 const NameAttrList** value); // type: "func"
258 const NameAttrList** value); // type: "func"
262 NameAttrList* value); // type: "func"
264 std::vector<NameAttrList>* value); // type: "list(func)"
|
| D | attr_value_util.h | 35 class NameAttrList; variable
69 void SetAttrValue(const NameAttrList& value, AttrValue* out);
88 void SetAttrValue(gtl::ArraySlice<NameAttrList> value, AttrValue* out);
|
| D | attr_value_util.cc | 137 const NameAttrList& func = a.func(); in AttrValueHash()
171 const NameAttrList& af = a.func(); in AreAttrValuesEqual()
172 const NameAttrList& bf = b.func(); in AreAttrValuesEqual()
217 string SummarizeFunc(const NameAttrList& func) { in SummarizeFunc()
577 void SetAttrValue(const NameAttrList& value, AttrValue* out) { in SetAttrValue()
581 void SetAttrValue(gtl::ArraySlice<NameAttrList> value, AttrValue* out) { in SetAttrValue()
607 for (const NameAttrList& func : val.list().func()) { in HasPlaceHolder()
635 for (NameAttrList& func : *value->mutable_list()->mutable_func()) { in SubstitutePlaceholders()
|
| D | attr_value.proto | 28 repeated NameAttrList func = 9; // "list(attr)"
46 NameAttrList func = 10;
61 message NameAttrList { message
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.h | 83 const NameAttrList& function, const tpu::TPUCompileMetadataProto metadata, in TpuCompileOpKernelCommon()
177 const NameAttrList& function,
212 NameAttrList function_;
|