Searched refs:kTPUReplicatedInput (Results 1 – 1 of 1) sorted by relevance
54 const char* const kTPUReplicatedInput = "TPUReplicatedInput"; variable88 arg_source_tensors.at(index).node->type_string() == kTPUReplicatedInput; in RewriteSubgraph()104 kTPUReplicatedInput) { in RewriteSubgraph()1581 if (n->type_string() == kTPUReplicatedInput) { in RemoveUnusedTPUReplicatedInputs()2628 in_edges[pos]->src()->type_string() == kTPUReplicatedInput) { in BuildTPUReplicateOps()