Searched defs:tpu_input_shapes (Results 1 – 1 of 1) sorted by relevance
402 std::map<int, InferredShape>& arg_shapes, EdgeShapes& tpu_input_shapes, in GetInputOutputInfo()652 const EdgeShapes& tpu_input_shapes, in GroupTensorsForInputPacking()746 Graph* graph, const string& cluster_name, EdgeShapes* tpu_input_shapes, in CreateConcatAndSplitNodesForInputTensor()1065 EdgeShapes* tpu_input_shapes, in InsertReshapeNodePairs()2198 EdgeShapes tpu_input_shapes; in OptimizeTpuInputOutputTensors() local