Searched defs:ExecutionInput (Results 1 – 1 of 1) sorted by relevance
63 explicit ExecutionInput(xla::Shape shape) : buffers_(std::move(shape)) { in ExecutionInput() function67 ExecutionInput(xla::Shape shape, xla::Shape host_shape) in ExecutionInput() function72 explicit ExecutionInput(ShapeTree<MaybeOwningDeviceMemory> buffers) in ExecutionInput() function77 ExecutionInput(ShapeTree<MaybeOwningDeviceMemory> buffers, in ExecutionInput() function