Searched defs:shape_to_output (Results 1 – 1 of 1) sorted by relevance
712 GroupedEdges shape_to_output; in GroupTensorsForOutputPacking() local926 GraphShapeInfo* tpu_inferred_info, GroupedEdges shape_to_output, in CreateConcatAndSplitNodesForOutputTensor()2255 GroupedEdges shape_to_output = in OptimizeTpuInputOutputTensors() local