Searched refs:XlaResourceKind (Results 1 – 5 of 5) sorted by relevance
35 enum class XlaResourceKind { enum44 XlaResourceKind resource_kind) in XlaResourceOpInfo()48 XlaResourceKind resource_kind() const { return resource_kind_; } in resource_kind()54 XlaResourceKind resource_kind_;
38 XlaResourceKind resource_kind) { in CreateResourceOpInfoMap()48 auto kVariable = XlaResourceKind::kVariable; in CreateResourceOpInfoMap()49 auto kStack = XlaResourceKind::kStack; in CreateResourceOpInfoMap()50 auto kTensorArray = XlaResourceKind::kTensorArray; in CreateResourceOpInfoMap()
78 if (op_info->resource_kind() != XlaResourceKind::kStack && in IsStackOrTensorArraySource()79 op_info->resource_kind() != XlaResourceKind::kTensorArray) in IsStackOrTensorArraySource()
243 return op_info && op_info->resource_kind() == XlaResourceKind::kStack; in IsStackOp()249 return op_info && op_info->resource_kind() == XlaResourceKind::kTensorArray; in IsTensorArrayOp()
1269 op_info && op_info->resource_kind() != XlaResourceKind::kVariable; in FindCompilationCandidates()