Searched defs:tpu_output_shapes (Results 1 – 1 of 1) sorted by relevance
660 EdgeShapes& tpu_output_shapes, in GroupTensorsForOutputPacking()875 Graph* graph, const string& cluster_name, EdgeShapes* tpu_output_shapes, in CreateConcatAndSplitNodesForOutputTensor()2166 EdgeShapes tpu_output_shapes; in OptimizeTpuInputOutputTensors() local