Searched refs:XlaResourceOpInfo (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_operation_table.cc | 21 /*static*/ absl::string_view XlaResourceOpInfo::XlaResourceOpKindToString( in XlaResourceOpKindToString()
33 static absl::flat_hash_map<absl::string_view, XlaResourceOpInfo>*
35 auto* result = new absl::flat_hash_map<absl::string_view, XlaResourceOpInfo>; in CreateResourceOpInfoMap()
40 result->insert({op, XlaResourceOpInfo(op_kind, resource_kind)}); in CreateResourceOpInfoMap()
116 static const absl::flat_hash_map<absl::string_view, XlaResourceOpInfo>&
118 static absl::flat_hash_map<absl::string_view, XlaResourceOpInfo>* in GetStaticResourceOpInfoMap()
123 const XlaResourceOpInfo* GetResourceOpInfoForOp(absl::string_view op) { in GetResourceOpInfoForOp()
124 const absl::flat_hash_map<absl::string_view, XlaResourceOpInfo>& op_infos = in GetResourceOpInfoForOp()
|
| D | resource_operation_table.h | 41 class XlaResourceOpInfo {
43 explicit XlaResourceOpInfo(XlaResourceOpKind op_kind, in XlaResourceOpInfo() function
60 const XlaResourceOpInfo* GetResourceOpInfoForOp(absl::string_view op);
|
| D | resource_util.cc | 75 const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(n.type_string()); in IsStackOrTensorArraySource()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | resource_operation_safety_analysis.cc | 113 const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(n.type_string()); in XlaResourceOpKindForNode()
149 XlaResourceOpInfo::XlaResourceOpKindToString(resource_op.second)); in ResourceOpToString()
245 XlaResourceOpInfo::XlaResourceOpKindToString(resource_op_kind), ")", "]"); in NodeToString()
|
| D | compilability_check_util.h | 241 const XlaResourceOpInfo* op_info = in IsStackOp()
247 const XlaResourceOpInfo* op_info = in IsTensorArrayOp()
|
| D | mark_for_compilation_pass.cc | 1266 const XlaResourceOpInfo* op_info = in FindCompilationCandidates()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2615 const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(n->type_string()); in ContainsResourceWriteOp()
2624 const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(n.op()); in ContainsResourceWriteOp()
|