Searched defs:xla_arg_shape (Results 1 – 1 of 1) sorted by relevance
50 xla::Shape* xla_arg_shape, in SetPerCoreArgShapes()322 xla::Shape xla_arg_shape; in BuildComputationArgumentDescriptions() local