Searched defs:per_core_output_shapes (Results 1 – 1 of 1) sorted by relevance
206 std::vector<std::vector<xla::Shape>>* per_core_output_shapes, in AddVariableUpdatesToCores()277 std::vector<std::vector<xla::Shape>>* per_core_output_shapes) { in ComputeOutputShapesForEachCore()