Searched defs:per_core_arg_shapes (Results 1 – 1 of 1) sorted by relevance
52 std::vector<std::vector<xla::Shape>>* per_core_arg_shapes) { in SetPerCoreArgShapes()258 std::vector<std::vector<xla::Shape>>* per_core_arg_shapes) { in BuildComputationArgumentDescriptions()379 std::vector<std::vector<xla::Shape>>* per_core_arg_shapes, in CompileTFFunctionToHlo()456 std::vector<std::vector<xla::Shape>>* per_core_arg_shapes) { in GetShardingInfo()