Searched defs:return_shape (Results 1 – 4 of 4) sorted by relevance
783 const Shape* return_shape = &shape; in GetSubshape() local794 const Shape* return_shape = &shape; in TryGetSubshape() local809 Shape* return_shape = shape; in GetMutableSubshape() local
944 Shape return_shape; in TransferToClient() local
1088 xla::Shape return_shape = xla::ShapeUtil::MakeTupleShape( in TEST() local
3044 const Shape& return_shape = callee.root_instruction()->shape(); in EmitThreadLocalCall() local