Searched defs:tpu_inferred_info (Results 1 – 1 of 1) sorted by relevance
401 Graph* graph, GraphShapeInfo& tpu_inferred_info, in GetInputOutputInfo()926 GraphShapeInfo* tpu_inferred_info, GroupedEdges shape_to_output, in CreateConcatAndSplitNodesForOutputTensor()1998 GraphShapeInfo* tpu_inferred_info) { in InferShapesWithResourceVar()2196 GraphShapeInfo tpu_inferred_info; in OptimizeTpuInputOutputTensors() local