Searched refs:op_kind (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | resource_operation_safety_analysis.cc | 284 absl::optional<XlaResourceOpKind> op_kind; in ComputeIncompatibleResourceOperationPairs() local
286 *n, flib_def, resource_ops_to_ignore, &op_kind)); in ComputeIncompatibleResourceOperationPairs()
304 if (op_kind) { in ComputeIncompatibleResourceOperationPairs()
306 if (IsEdgeSafe(incoming_op.second, *op_kind)) { in ComputeIncompatibleResourceOperationPairs()
314 << " -> " << NodeToString(*n, *op_kind); in ComputeIncompatibleResourceOperationPairs()
319 resource_op_set->Add({n->id(), *op_kind}); in ComputeIncompatibleResourceOperationPairs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_operation_table.h | 43 explicit XlaResourceOpInfo(XlaResourceOpKind op_kind, in XlaResourceOpInfo() argument
45 : op_kind_(op_kind), resource_kind_(resource_kind) {} in XlaResourceOpInfo()
50 static absl::string_view XlaResourceOpKindToString(XlaResourceOpKind op_kind);
|
| D | resource_operation_table.cc | 22 XlaResourceOpKind op_kind) { in XlaResourceOpKindToString() argument
23 switch (op_kind) { in XlaResourceOpKindToString()
37 auto add = [&](absl::string_view op, XlaResourceOpKind op_kind, in CreateResourceOpInfoMap()
40 result->insert({op, XlaResourceOpInfo(op_kind, resource_kind)}); in CreateResourceOpInfoMap()
|
| /external/mesa3d/src/gallium/drivers/r600/sb/ |
| D | sb_peephole.cpp | 124 int op_kind = (flags & AF_PRED) ? 1 : in optimize_cc_op2() local
178 switch(op_kind) { in optimize_cc_op2()
|
| /external/python/cpython3/Parser/ |
| D | asdl.py | 239 op_kind = TokenKind.operator_table[c]
242 yield Token(op_kind, c, lineno)
|