Searched refs:resource_ops_to_ignore (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | resource_operation_safety_analysis.cc | 117 const std::function<Status(const Node&, bool*)>& resource_ops_to_ignore, in XlaResourceOpKindForNode() argument 120 if (resource_ops_to_ignore) { in XlaResourceOpKindForNode() 121 TF_RETURN_IF_ERROR(resource_ops_to_ignore(n, &should_ignore)); in XlaResourceOpKindForNode() 268 const std::function<Status(const Node&, bool*)>& resource_ops_to_ignore, in ComputeIncompatibleResourceOperationPairs() argument 286 *n, flib_def, resource_ops_to_ignore, &op_kind)); in ComputeIncompatibleResourceOperationPairs()
|
D | resource_operation_safety_analysis.h | 69 const std::function<Status(const Node&, bool*)>& resource_ops_to_ignore,
|
D | xla_cluster_util.h | 81 const std::function<Status(const Node&, bool*)>& resource_ops_to_ignore,
|
D | xla_cluster_util.cc | 231 const std::function<Status(const Node&, bool*)>& resource_ops_to_ignore, in AdjustCycleDetectionGraphForResourceOps() argument 235 *graph, flib_def, resource_ops_to_ignore, &unsafe_deps)); in AdjustCycleDetectionGraphForResourceOps()
|