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