Searched refs:kTPUPartitionedInput (Results 1 – 2 of 2) sorted by relevance
92 const char kTPUPartitionedInput[] = "TPUPartitionedInput"; variable2001 if (input_node->type_string() == kTPUPartitionedInput) { in AssignArgsAndRetvalsToCores()2917 if (e->src()->type_string() == kTPUPartitionedInput) { in BuildExecuteNodes()3153 if (edge->src()->type_string() == kTPUPartitionedInput) { in BuildExecuteNodes()3182 } else if (edge->src()->type_string() == kTPUPartitionedInput && in BuildExecuteNodes()4461 if (n->type_string() == kTPUPartitionedInput) graph->RemoveNode(n); in Run()
57 const char* const kTPUPartitionedInput = "TPUPartitionedInput"; variable1595 e_in->src()->type_string() == kTPUPartitionedInput) in RemoveUnusedTPUReplicatedInputs()