Home
last modified time | relevance | path

Searched refs:NumOutputsToHost (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h216 int64 NumOutputsToHost() const { in NumOutputsToHost() function
Ddistributed_tpu_rewrite_pass.cc4023 if (replicate_node.num_outputs() != params_info->NumOutputsToHost()) { in GetIOTypes()
4026 params_info->NumOutputsToHost(), " outputs, got ", in GetIOTypes()