Searched defs:compile_node (Results 1 – 4 of 4) sorted by relevance
| /external/yapf/third_party/yapf_third_party/_ylib2to3/ |
| D | patcomp.py | 70 def compile_node(self, node): member in PatternCompiler
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 114 Status ExtractExecuteNodeInfo(const Node* compile_node, const Graph& graph, in ExtractExecuteNodeInfo()
460 const auto compile_node = node_it->second; in AddReshardOp() local
|
| D | tpu_embedding_software_deduplication_rewrite_pass.cc | 363 const Node* compile_node = in ValidateAndGetTPUEmbeddingConfiguration() local
411 const Node* compile_node = (old_activations_node != nullptr) in AddRecvDeduplicationDataNode() local
|
| D | distributed_tpu_rewrite_pass.cc | 2606 Node** compile_node, int64_t autotuner_thresh) { in BuildCompileNode()
2836 Node* compile_node, Node* key_placeholder_node, Graph* graph) { in ConnectHostComputeNodes()
3248 Node* compile_node, const std::vector<Node*>& variable_reads, in BuildExecuteNodes()
4571 Node* replicate_node, Node* compile_node, in BuildCompilationStatusReturnNodes()
4822 Node* compile_node; in RewriteTPUReplicateNode() local
|