Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for RemoveFromXlaCluster (0.3 sec)

  1. tensorflow/compiler/jit/partially_decluster_pass.cc

                  VLOG(3) << "Declustering Const with cross-device control input "
                          << n->name();
                  RemoveFromXlaCluster(n);
                  break;
                }
              }
            } else {
              VLOG(3) << "Declustering must-be-constant node " << n->name();
              RemoveFromXlaCluster(n);
            }
          }
        }
      }
    
      return absl::OkStatus();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_cluster_util.cc

             std::find(node.output_types().begin(), node.output_types().end(),
                       DT_RESOURCE) != node.output_types().end();
    }
    
    void RemoveFromXlaCluster(NodeDef* node_def) {
      node_def->mutable_attr()->erase(kXlaClusterAttr);
    }
    
    void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); }
    
    namespace {
    typedef xla_config_registry::XlaGlobalJitLevel XlaGlobalJitLevel;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
Back to top